This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch unity-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 11c73a2ea6 Merge remote-tracking branch 'apache-upstream/main' into
unity-staging
add d2a9f39f34 [Fix] Fix Build with RCCL (#15863)
add 8954139bf6 [Unity] Add R.quantize and R.dequantize ops (#15849)
add 7d53953adc [Unity] Check the PassContext within RewriteCUDAGraph
transform (#15861)
add 9f0ac49a04 [CMake] Prefer a static NCCL (#15869)
add 6d2b44dbbf [Bugfix] Fix Disco-CUDAGraph Integration (#15870)
add b486210b1a [Disco] Add -lrt to TVM runtime for NCCL (#15876)
add 2e30dbe0ff [Unity][Fix] Remove duplicated words from comments, NFC
(#15875)
add 8a2ca3405b [Unity] Use PrimValue as offset in R.tril and R.triu
(#15783)
add 59ec81f125 [Unity][MSC][pre M1.2] Reconstruct codegen (#15813)
add 7c4c2c2f42 [Disco][Fix] Remove Dependency to PyTest (#15886)
add 969e31a978 [Unity][NN] Enhance ReLU and GELU support (#15885)
add 646d3a2c5e [TFLite][Frontend] Support quantized Pow (#15798)
add a38053e92e [Script] Fix miscs of make_notes.py (#15859)
add b8abff9789 [VM] Memory Manager moved up to runtime (#15833)
add 7a1f7d0b5a [Docker] Upgrade oneflow to v0.8.0 (#15862)
add e754bc2325 [Relay][Bugfix] Fix conv transpose with default strides in
ONNX frontend (#15868)
add 958c27123a [Fix] Remove duplicated words from comments, NFC (#15873)
add e89b39e55b Bump pillow from 9.2.0 to 10.0.1 in /apps/microtvm (#15865)
add 6f04fdab5a Bump pillow from 9.3.0 to 10.0.1 in /apps/microtvm/cmsisnn
(#15866)
add ec92ea960d Bump pillow from 9.3.0 to 10.0.1 in /apps/microtvm/ethosu
(#15867)
add fa4aeee64e [CI] Update ci-gpu image (#15836)
add dc53a6c29a [Arith] Simplify the result of non-divisible floordiv
(#15881)
new b9a02b1e84 Merge remote-tracking branch 'apache-upstream/main' into
apache-upstream-unity
The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
CMakeLists.txt | 6 +-
apps/android_camera/app/src/main/jni/tvm_runtime.h | 1 +
apps/android_deploy/app/src/main/jni/tvm_runtime.h | 1 +
apps/android_rpc/app/src/main/jni/tvm_runtime.h | 1 +
apps/bundle_deploy/runtime.cc | 1 +
apps/howto_deploy/tvm_runtime_pack.cc | 1 +
apps/microtvm/cmsisnn/requirements.txt | 117 ++---
apps/microtvm/ethosu/requirements.txt | 117 ++---
apps/microtvm/poetry.lock | 131 +++--
apps/microtvm/pyproject.toml | 2 +-
ci/jenkins/docker-images.ini | 2 +-
cmake/utils/FindNCCL.cmake | 5 +-
docker/install/ubuntu_install_oneflow.sh | 2 +-
docs/Doxyfile | 2 +-
docs/conf.py | 4 +-
golang/src/tvm_runtime_pack.cc | 1 +
include/tvm/relax/attrs/{search.h => qdq.h} | 32 +-
include/tvm/relax/transform.h | 2 +-
include/tvm/runtime/device_api.h | 8 +
include/tvm/runtime/logging.h | 2 +-
.../tvm/runtime/{vm => memory}/memory_manager.h | 45 +-
include/tvm/runtime/ndarray.h | 2 +-
include/tvm/runtime/packed_func.h | 2 +-
include/tvm/runtime/vm/vm.h | 9 +-
include/tvm/tir/expr.h | 2 +-
include/tvm/tir/stmt.h | 2 +-
python/tvm/contrib/msc/core/codegen/__init__.py | 1 +
python/tvm/contrib/msc/core/codegen/codegen.py | 53 +-
python/tvm/contrib/msc/core/codegen/sources.py | 184 +++++++
python/tvm/contrib/msc/core/ir/graph.py | 50 +-
python/tvm/contrib/msc/core/ir/translate.py | 189 ++++++-
python/tvm/contrib/msc/core/transform/pattern.py | 146 ++++++
python/tvm/contrib/msc/core/transform/transform.py | 33 +-
python/tvm/contrib/msc/core/utils/expr.py | 19 +-
python/tvm/contrib/msc/core/utils/file.py | 80 ++-
python/tvm/contrib/msc/core/utils/info.py | 40 ++
python/tvm/contrib/msc/core/utils/namespace.py | 2 +
.../contrib/msc/framework/torch/codegen/codegen.py | 2 +-
.../contrib/msc/framework/tvm/codegen/codegen.py | 2 +-
python/tvm/exec/disco_worker.py | 37 +-
python/tvm/relax/frontend/nn/__init__.py | 3 +
python/tvm/relax/frontend/nn/modules.py | 18 +-
python/tvm/relax/frontend/nn/op.py | 24 +-
python/tvm/relax/op/__init__.py | 1 +
python/tvm/relax/op/_op_gradient.py | 2 +-
python/tvm/relax/op/create.py | 10 +-
python/tvm/relax/op/linear_algebra.py | 4 +-
python/tvm/relax/op/qdq.py | 88 ++++
python/tvm/relax/struct_info.py | 4 +-
python/tvm/relax/training/utils.py | 2 +-
.../tvm/relax/transform/legalize_ops/__init__.py | 1 +
python/tvm/relax/transform/legalize_ops/create.py | 5 +-
python/tvm/relax/transform/legalize_ops/qdq.py | 114 ++++
python/tvm/relax/transform/transform.py | 8 +-
python/tvm/relax/vm_build.py | 36 +-
python/tvm/relay/frontend/oneflow.py | 12 +-
python/tvm/relay/frontend/onnx.py | 4 +-
python/tvm/relay/frontend/tflite.py | 2 -
python/tvm/relay/op/contrib/clml.py | 2 +-
python/tvm/relay/transform/memory_plan.py | 2 +-
python/tvm/runtime/disco/process_pool.py | 2 +-
python/tvm/runtime/ndarray.py | 2 +-
python/tvm/script/ir_builder/relax/ir.py | 4 +
python/tvm/te/hybrid/__init__.py | 2 +-
python/tvm/te/schedule.py | 2 +-
python/tvm/testing/__init__.py | 2 +-
python/tvm/testing/disco.py | 53 --
python/tvm/topi/arm_cpu/qnn.py | 2 +-
src/arith/const_fold.h | 4 +-
src/arith/iter_affine_map.cc | 9 +
src/arith/product_normal_form.h | 4 +-
src/contrib/msc/core/codegen/base_codegen.h | 43 +-
src/contrib/msc/core/codegen/code_stack.cc | 175 ++++---
src/contrib/msc/core/codegen/code_stack.h | 484 +++++++++--------
src/contrib/msc/core/codegen/codegen_utils.h | 59 +--
src/contrib/msc/core/codegen/py_codegen.h | 59 +--
src/contrib/msc/core/ir/graph.cc | 25 +
src/contrib/msc/core/ir/graph.h | 2 +
src/contrib/msc/core/ir/graph_builder.cc | 156 +++++-
src/contrib/msc/core/ir/graph_builder.h | 49 +-
src/contrib/msc/core/printer/msc_base_printer.cc | 21 +-
src/contrib/msc/core/printer/msc_base_printer.h | 21 +-
.../tvm/config.h => core/printer/msc_doc.cc} | 53 +-
src/contrib/msc/core/printer/msc_doc.h | 160 ++++++
src/contrib/msc/core/printer/print_utils.cc | 24 +
src/contrib/msc/core/printer/print_utils.h | 67 ++-
src/contrib/msc/core/printer/python_printer.cc | 8 +
src/contrib/msc/core/printer/python_printer.h | 7 +-
src/contrib/msc/core/transform/layout_utils.cc | 2 +-
src/contrib/msc/core/transform/set_expr_layout.cc | 250 ++++++---
src/contrib/msc/core/transform/set_expr_name.cc | 236 ++++++---
src/contrib/msc/core/utils.cc | 71 ++-
src/contrib/msc/core/utils.h | 24 +-
src/contrib/msc/framework/torch/codegen.cc | 44 +-
src/contrib/msc/framework/torch/torch_opcode.cc | 156 +++---
src/contrib/msc/framework/torch/torch_opcode.h | 2 +-
src/contrib/msc/framework/tvm/codegen.cc | 83 ++-
src/contrib/msc/framework/tvm/relax_opcode.cc | 362 ++++++++-----
src/contrib/msc/framework/tvm/relax_opcode.h | 2 +-
src/relax/op/op_common.h | 60 +++
src/relax/op/tensor/create.cc | 31 +-
src/relax/op/tensor/create.h | 12 +
src/relax/op/tensor/linear_algebra.h | 2 +-
src/relax/op/tensor/qdq.cc | 201 ++++++++
src/relax/op/tensor/qdq.h | 61 +++
src/relax/training/utils.h | 2 +-
src/relax/transform/fuse_ops.cc | 2 +-
src/relax/transform/rewrite_cuda_graph.cc | 10 +-
src/relax/transform/static_plan_block_memory.cc | 2 +-
src/relay/backend/vm/compiler.h | 2 +-
src/relay/collage/partition_rule.h | 2 +-
src/relay/transforms/combine_parallel_op_batch.h | 2 +-
src/runtime/c_runtime_api.cc | 16 +-
src/runtime/crt/aot_executor/aot_executor.c | 2 +-
src/runtime/disco/nccl/nccl.cc | 19 +-
src/runtime/graph_executor/graph_executor.cc | 3 +-
src/runtime/graph_executor/graph_executor.h | 4 +
src/runtime/{vm => memory}/memory_manager.cc | 59 ++-
src/runtime/{vm => memory}/naive_allocator.h | 29 +-
src/runtime/{vm => memory}/pooled_allocator.h | 20 +-
src/runtime/vm/vm.cc | 14 +-
src/target/source/ptx.h | 2 +-
src/tir/analysis/control_flow_graph.h | 2 +-
src/tir/schedule/error.h | 2 +-
src/tir/transforms/unroll_loop.cc | 2 +-
.../runtime/{vm => memory}/memory_manager_tests.cc | 39 +-
.../contrib/test_arm_compute_lib/infrastructure.py | 2 +-
.../test_hexagon/conv2d/test_conv2d_conv2d.md | 2 +-
.../test_hexagon/test_benchmark_elemwise_add.py | 2 +-
tests/python/contrib/test_msc/test_graph_build.py | 8 +-
.../contrib/test_msc/test_translate_relax.py | 8 +-
.../contrib/test_msc/test_translate_relay.py | 14 +-
.../contrib/test_msc/test_translate_torch.py | 39 +-
tests/python/frontend/coreml/test_forward.py | 4 +-
tests/python/frontend/onnx/test_forward.py | 9 +
tests/python/frontend/tflite/test_forward.py | 13 +-
tests/python/relax/test_frontend_nn_modules.py | 36 ++
tests/python/relax/test_frontend_nn_op.py | 2 +
tests/python/relax/test_op_create.py | 8 +
tests/python/relax/test_op_qdq.py | 72 +++
tests/python/relax/test_transform_fold_constant.py | 2 +-
.../relax/test_transform_legalize_ops_qdq.py | 572 +++++++++++++++++++++
.../relax/test_transform_rewrite_cuda_graph.py | 27 +
tests/python/relay/test_pass_plan_devices.py | 2 +-
.../python/unittest/test_arith_iter_affine_map.py | 34 ++
.../unittest/test_tir_transform_remove_no_op.py | 2 +-
tests/scripts/release/make_notes.py | 4 +-
web/src/environment.ts | 2 +-
148 files changed, 4449 insertions(+), 1428 deletions(-)
copy include/tvm/relax/attrs/{search.h => qdq.h} (55%)
rename include/tvm/runtime/{vm => memory}/memory_manager.h (82%)
create mode 100644 python/tvm/contrib/msc/core/codegen/sources.py
create mode 100644 python/tvm/relax/op/qdq.py
create mode 100644 python/tvm/relax/transform/legalize_ops/qdq.py
delete mode 100644 python/tvm/testing/disco.py
copy src/contrib/msc/{framework/tvm/config.h => core/printer/msc_doc.cc} (53%)
create mode 100644 src/contrib/msc/core/printer/msc_doc.h
create mode 100644 src/relax/op/tensor/qdq.cc
create mode 100644 src/relax/op/tensor/qdq.h
rename src/runtime/{vm => memory}/memory_manager.cc (76%)
rename src/runtime/{vm => memory}/naive_allocator.h (78%)
rename src/runtime/{vm => memory}/pooled_allocator.h (88%)
rename tests/cpp/runtime/{vm => memory}/memory_manager_tests.cc (86%)
create mode 100644 tests/python/relax/test_op_qdq.py
create mode 100644 tests/python/relax/test_transform_legalize_ops_qdq.py
