This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 048994b [Bug][Meta Schedule] Fix Infinite Loop Caused When Calling
Methods Not Overrided In PyClass. (#9451)
add e7024fb [PyTorch]Add PyTorchTVM: compile torchscript to tvm and
export as pytorch_op (#8777)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 2 +
apps/{tf_tvmdsoop => pt_tvmdsoop}/CMakeLists.txt | 10 +-
.../prepare_and_test_pt_tvm_class.sh} | 19 +-
apps/pt_tvmdsoop/tests/test_torch_compile_cpu.py | 68 ++
apps/pt_tvmdsoop/tests/test_torch_compile_gpu.py | 63 ++
apps/pt_tvmdsoop/tests/test_torch_graph_module.py | 129 ++++
apps/pt_tvmdsoop/tests/test_torch_script.py | 116 ++++
apps/pt_tvmdsoop/tests/test_torch_vm_module.py | 122 ++++
apps/pt_tvmdsoop/tests/test_trace_tvm_module.py | 58 ++
cmake/config.cmake | 3 +
cmake/modules/LibInfo.cmake | 1 +
cmake/modules/contrib/PT_TVMDSOOP.cmake | 59 ++
python/tvm/contrib/torch/__init__.py | 51 ++
python/tvm/contrib/torch/module.py | 121 ++++
python/tvm/contrib/torch/pytorch_tvm.py | 249 ++++++++
src/contrib/torch/pt_call_tvm/tvm_class.cc | 686 +++++++++++++++++++++
src/contrib/torch/utils.h | 264 ++++++++
17 files changed, 2012 insertions(+), 9 deletions(-)
copy apps/{tf_tvmdsoop => pt_tvmdsoop}/CMakeLists.txt (84%)
copy apps/{tf_tvmdsoop/prepare_and_test_tfop_module.sh =>
pt_tvmdsoop/prepare_and_test_pt_tvm_class.sh} (72%)
mode change 100644 => 100755
create mode 100644 apps/pt_tvmdsoop/tests/test_torch_compile_cpu.py
create mode 100644 apps/pt_tvmdsoop/tests/test_torch_compile_gpu.py
create mode 100644 apps/pt_tvmdsoop/tests/test_torch_graph_module.py
create mode 100644 apps/pt_tvmdsoop/tests/test_torch_script.py
create mode 100644 apps/pt_tvmdsoop/tests/test_torch_vm_module.py
create mode 100644 apps/pt_tvmdsoop/tests/test_trace_tvm_module.py
create mode 100644 cmake/modules/contrib/PT_TVMDSOOP.cmake
create mode 100644 python/tvm/contrib/torch/__init__.py
create mode 100644 python/tvm/contrib/torch/module.py
create mode 100644 python/tvm/contrib/torch/pytorch_tvm.py
create mode 100644 src/contrib/torch/pt_call_tvm/tvm_class.cc
create mode 100644 src/contrib/torch/utils.h
