projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0ce4d30) | patch
Move fx2trt and oss_acc_tracer to oss (#63101)
authorShiyan Deng <dsy842974287@fb.com>
Sun, 15 Aug 2021 18:52:20 +0000 (11:52 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Sun, 15 Aug 2021 18:53:36 +0000 (11:53 -0700)
commit8e0998ca70ed9b878d0f11d435bbe51bb95aa1ca
tree6ad6ed64959c1476da4a3fe18ce143c0528982fbtree | snapshot
parent0ce4d30c4466ad9b0402e221329070a97be545eacommit | diff
Move fx2trt and oss_acc_tracer to oss (#63101)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63101

Move internal fx2trt to torch/fx/experimental/fx2trt and merge the two TRT interpreter we have right now. cc: mortzur as this might affect uru exporting script.

Move oss_acc_tracer to torch/fx/experimental/fx_acc.

Test Plan: CI

Reviewed By: jerryzh168

Differential Revision: D30257909

fbshipit-source-id: 4e374965fbf88d72e91844d9e9b6ff9b98f467d1
12 files changed:
test/test_fx_experimental.py diff | blob | history
torch/fx/experimental/fx2trt/converters/__init__.py diff | blob | history
torch/fx/experimental/fx2trt/converters/acc_ops_converters.py [new file with mode: 0644] blob
torch/fx/experimental/fx2trt/converters/add.py diff | blob | history
torch/fx/experimental/fx2trt/converters/mul.py diff | blob | history
torch/fx/experimental/fx2trt/fx2trt.py diff | blob | history
torch/fx/experimental/fx_acc/__init__.py [new file with mode: 0644] blob
torch/fx/experimental/fx_acc/acc_normalizer.py [new file with mode: 0644] blob
torch/fx/experimental/fx_acc/acc_ops.py [new file with mode: 0644] blob
torch/fx/experimental/fx_acc/acc_tracer.py [new file with mode: 0644] blob
torch/fx/experimental/fx_acc/acc_utils.py [new file with mode: 0644] blob
torch/fx/experimental/graph_manipulation.py diff | blob | history
Domain: Machine Learning / ML Framework;