You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by co...@apache.org on 2021/03/17 16:39:26 UTC

[tvm] branch main updated (c55608f -> 60ff0c7)

This is an automated email from the ASF dual-hosted git repository.

comaniac pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.


    from c55608f  [SPIRV] Declare int64 capability by default (#7681)
     add 60ff0c7  [Runtime] Extend Graph Runtime To Support Cuda Graph Launch (#7616)

No new revisions were added by this update.

Summary of changes:
 CMakeLists.txt                                     |   1 +
 cmake/config.cmake                                 |   3 +
 cmake/modules/CUDA.cmake                           |  11 ++
 .../cuda_graph}/__init__.py                        |   0
 .../tvm/contrib/cuda_graph/cuda_graph_runtime.py   | 134 ++++++++++++++++++++
 python/tvm/contrib/nvcc.py                         |  12 ++
 python/tvm/testing.py                              |  19 +++
 .../graph/cuda_graph/graph_runtime_cuda_graph.cc   | 135 +++++++++++++++++++++
 src/runtime/graph/graph_runtime_factory.cc         |  33 +++++
 src/runtime/graph/graph_runtime_factory.h          |   8 ++
 .../unittest/test_runtime_graph_cuda_graph.py      |  65 +++++++---
 .../test_runtime_module_based_interface.py         |  30 +++++
 12 files changed, 434 insertions(+), 17 deletions(-)
 copy python/tvm/{topi/arm_cpu/cortex_m7/micro_kernel => contrib/cuda_graph}/__init__.py (100%)
 create mode 100644 python/tvm/contrib/cuda_graph/cuda_graph_runtime.py
 create mode 100644 src/runtime/graph/cuda_graph/graph_runtime_cuda_graph.cc
 copy jvm/core/src/test/scripts/test_graph_runtime.py => tests/python/unittest/test_runtime_graph_cuda_graph.py (55%)