This is an automated email from the ASF dual-hosted git repository.
ruihangl pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 472b2fcd22 [Relax] ONNX frontend using relax softplus operator (#18242)
add 2012d55caf [Relax] Add Python function support and BasePyModule for
PyTorch integration (#18229)
No new revisions were added by this update.
Summary of changes:
python/tvm/ir/module.py | 1 +
python/tvm/relax/__init__.py | 3 +
python/tvm/relax/base_py_module.py | 385 +++++++++++++++++++++++++
python/tvm/script/parser/core/entry.py | 69 +++++
python/tvm/script/parser/core/parser.py | 35 +++
python/tvm/script/parser/ir/__init__.py | 3 +-
python/tvm/script/parser/ir/entry.py | 94 +++++-
python/tvm/script/parser/ir/parser.py | 88 +++++-
src/ir/function.cc | 5 +
tests/python/relax/test_base_py_module.py | 206 +++++++++++++
tests/python/relax/test_dlpack_integration.py | 296 +++++++++++++++++++
tests/python/relax/test_pytorch_integration.py | 380 ++++++++++++++++++++++++
tests/python/relax/test_tvmscript_pyfunc.py | 268 +++++++++++++++++
13 files changed, 1826 insertions(+), 7 deletions(-)
create mode 100644 python/tvm/relax/base_py_module.py
create mode 100644 tests/python/relax/test_base_py_module.py
create mode 100644 tests/python/relax/test_dlpack_integration.py
create mode 100644 tests/python/relax/test_pytorch_integration.py
create mode 100644 tests/python/relax/test_tvmscript_pyfunc.py
