This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from ffd5f70 Revert "[TE][Fix] Comparison of the output tensor (#9829)"
(#10540)
add fe33ed6 RFC: initial stab at TorchScript fallback (#7401)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 2 +
cmake/config.cmake | 5 +
.../modules/contrib/LibTorch.cmake | 21 +--
python/tvm/relay/op/contrib/__init__.py | 1 +
.../__init__.py => relay/op/contrib/libtorch.py} | 28 +++-
.../backend/contrib/libtorch/libtorch_codegen.cc | 138 +++++++++++++++++
src/relay/transforms/type_infer.cc | 48 ++++--
src/runtime/contrib/libtorch/libtorch_runtime.cc | 168 +++++++++++++++++++++
tests/python/contrib/test_libtorch_ops.py | 88 +++++++++++
9 files changed, 468 insertions(+), 31 deletions(-)
copy python/tvm/contrib/debugger/debug_runtime.py =>
cmake/modules/contrib/LibTorch.cmake (64%)
copy python/tvm/{meta_schedule/task_scheduler/__init__.py =>
relay/op/contrib/libtorch.py} (57%)
create mode 100644 src/relay/backend/contrib/libtorch/libtorch_codegen.cc
create mode 100644 src/runtime/contrib/libtorch/libtorch_runtime.cc
create mode 100644 tests/python/contrib/test_libtorch_ops.py
