You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by zh...@apache.org on 2020/10/19 23:03:20 UTC
[incubator-tvm] branch main updated (28e9ab7 -> af8636a)
This is an automated email from the ASF dual-hosted git repository.
zhic pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 28e9ab7 Fix the Type bug in ConvertSSA. (#6709)
add af8636a [BYOC][TensorRT] TensorRT BYOC integration (#6395)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 3 +
cmake/config.cmake | 10 +
cmake/modules/contrib/TensorRT.cmake | 54 ++
docs/deploy/index.rst | 1 +
docs/deploy/tensorrt.rst | 297 ++++++
python/tvm/relay/op/contrib/__init__.py | 1 +
python/tvm/relay/op/contrib/tensorrt.py | 769 ++++++++++++++++
src/relay/backend/contrib/tensorrt/codegen.cc | 240 +++++
src/runtime/contrib/tensorrt/tensorrt_builder.cc | 222 +++++
src/runtime/contrib/tensorrt/tensorrt_builder.h | 159 ++++
src/runtime/contrib/tensorrt/tensorrt_logger.h | 78 ++
src/runtime/contrib/tensorrt/tensorrt_ops.cc | 1070 ++++++++++++++++++++++
src/runtime/contrib/tensorrt/tensorrt_ops.h | 207 +++++
src/runtime/contrib/tensorrt/tensorrt_runtime.cc | 312 +++++++
src/runtime/contrib/tensorrt/tensorrt_utils.h | 74 ++
tests/python/contrib/test_tensorrt.py | 905 ++++++++++++++++++
tests/scripts/task_config_build_gpu.sh | 1 +
17 files changed, 4403 insertions(+)
create mode 100644 cmake/modules/contrib/TensorRT.cmake
create mode 100644 docs/deploy/tensorrt.rst
create mode 100644 python/tvm/relay/op/contrib/tensorrt.py
create mode 100644 src/relay/backend/contrib/tensorrt/codegen.cc
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_builder.cc
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_builder.h
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_logger.h
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_ops.cc
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_ops.h
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_runtime.cc
create mode 100644 src/runtime/contrib/tensorrt/tensorrt_utils.h
create mode 100644 tests/python/contrib/test_tensorrt.py