This is an automated email from the ASF dual-hosted git repository. syfeng pushed a commit to branch test_all_cases_on_unity in repository https://gitbox.apache.org/repos/asf/tvm.git
commit d742c64023c0771b68b782573d2b6da774727632 Merge: c57407282c b5b980e33a Author: Siyuan Feng <[email protected]> AuthorDate: Tue Dec 12 10:05:42 2023 +0800 Merge commit 'b5b980e33a6af22793279d5bb85cc45c2b525799' into test_all_cases_on_unity 3rdparty/cutlass_fpA_intB_gemm | 2 +- 3rdparty/flashinfer | 2 +- {src => include/tvm}/runtime/disco/builtin.h | 38 +- .../tvm/runtime/disco/disco_worker.h | 71 +--- include/tvm/runtime/disco/session.h | 20 +- .../tvm}/runtime/relax_vm/ndarray_cache_support.h | 39 +- python/tvm/contrib/msc/core/ir/graph.py | 76 ++++ python/tvm/contrib/msc/core/runtime/runner.py | 286 +++++++++++---- python/tvm/contrib/msc/core/tools/__init__.py | 1 + python/tvm/contrib/msc/core/tools/prune/method.py | 6 +- python/tvm/contrib/msc/core/tools/prune/pruner.py | 149 +++++--- python/tvm/contrib/msc/core/tools/tool.py | 256 +++++++------ .../tvm/tools => core/tools/track}/__init__.py | 5 +- python/tvm/contrib/msc/core/tools/track/method.py | 102 ++++++ python/tvm/contrib/msc/core/tools/track/tracker.py | 185 ++++++++++ python/tvm/contrib/msc/core/utils/info.py | 37 +- python/tvm/contrib/msc/core/utils/log.py | 21 +- python/tvm/contrib/msc/core/utils/message.py | 23 ++ .../msc/framework/tensorflow/runtime/runner.py | 2 +- .../msc/framework/tensorflow/tools/__init__.py | 1 + .../tools => tensorflow/tools/track}/__init__.py | 4 +- .../framework/tensorflow/tools/track/tracker.py | 55 +++ .../msc/framework/tensorrt/tools/__init__.py | 1 + .../tools => tensorrt/tools/track}/__init__.py | 4 +- .../msc/framework/tensorrt/tools/track/tracker.py | 159 ++++++++ .../msc/framework/torch/frontend/translate.py | 2 +- .../contrib/msc/framework/torch/tools/__init__.py | 1 + .../{tvm/tools => torch/tools/track}/__init__.py | 4 +- .../msc/framework/torch/tools/track/tracker.py | 55 +++ .../contrib/msc/framework/tvm/tools/__init__.py | 1 + .../framework/tvm/tools/{ => track}/__init__.py | 4 +- .../msc/framework/tvm/tools/track/tracker.py | 155 ++++++++ python/tvm/contrib/msc/pipeline/manager.py | 112 ++++-- python/tvm/dlight/gpu/utils.py | 2 +- python/tvm/relax/expr.py | 5 + python/tvm/relax/frontend/nn/core.py | 2 + python/tvm/relax/frontend/nn/modules.py | 30 +- python/tvm/relax/frontend/nn/op.py | 39 ++ python/tvm/relax/frontend/nn/spec.py | 2 +- python/tvm/relax/transform/__init__.py | 1 + .../relax/transform/optimize_layout_transform.py | 37 +- .../relax/transform/remove_redundant_reshape.py | 37 +- python/tvm/relax/transform/transform.py | 10 + python/tvm/runtime/disco/process_pool.py | 16 +- python/tvm/runtime/disco/session.py | 3 +- src/contrib/msc/core/ir/graph.cc | 17 + src/contrib/msc/core/ir/graph.h | 2 +- src/contrib/msc/framework/tensorrt/codegen.cc | 37 +- src/contrib/msc/framework/tensorrt/codegen.h | 4 + src/relax/transform/canonicalize_bindings.cc | 45 ++- src/relax/transform/fuse_tir.cc | 2 - src/relax/transform/inline_functions.cc | 228 ++++++++++++ src/relax/transform/normalize.cc | 33 +- src/runtime/cuda/cuda_device_api.cc | 8 + src/runtime/disco/bcast_session.h | 3 +- src/runtime/disco/builtin.cc | 5 +- src/runtime/disco/{worker.cc => disco_worker.cc} | 8 +- src/runtime/disco/disco_worker_thread.h | 83 +++++ src/runtime/disco/loader.cc | 89 ++++- src/runtime/disco/nccl/nccl.cc | 5 +- src/runtime/disco/process_session.cc | 9 +- src/runtime/disco/session.cc | 3 +- src/runtime/disco/threaded_session.cc | 3 +- src/runtime/disco/utils.h | 32 +- src/runtime/relax_vm/ndarray_cache_support.cc | 161 ++++---- tests/python/contrib/test_msc/test_tools.py | 102 ++++-- tests/python/dlight/test_gpu_reduction.py | 343 ++++++++++------- tests/python/relax/test_inline_functions.py | 404 +++++++++++++++++++++ .../relax/test_transform_canonicalize_bindings.py | 163 ++++++++- .../test_transform_inline_private_functions.py | 105 ++++++ 70 files changed, 3097 insertions(+), 860 deletions(-)
