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 5828f1e9ee [Unity] Add a Standalone VM Version Number (#15254)
add 5dc25afc87 [microNPU][ETHOSU] Add Vela's logic to select configuration
block (#15186)
add 1d4829e430 [Android] ndk static build (#15215)
add 516c56b46a [Web] Increase default EMCC compilation total memory size
(#15218)
add 8f9f605dd5 [ARITH] Enhance buffer shape bound deduction to include
offset (#15228)
add 23fb568521 [CMAKE] Add Vulkan header for Android (#15229)
add c928852d59 [#15157][Rust][Doc] Re-enable the Rust documentation build
(#15213)
add 0bb390b272 [UnitTest][NVPTX] Avoid cascading failures from CUDA
postproc (#15136)
add 88701dc82a [Miscs] Enhance script about make release notes (#15234)
add 2f7c097594 [TIR] Allow VerifyWellFormed to accept IRModule (#15247)
add 73a62f647f [TIR] Preserve AllocateNode::annotations (#15242)
add 81463d79c0 [TIR][Schedule] Scoped CacheRead/Write producing compact
region (#15236)
add 3a33771494 [TVMScript] Handle parsing of PrimFunc calls with non-void
return (#15239)
add d9d6a88a0a [QNN] Support Dequantize to "float16" and Quantize to
"uint16" (#15235)
add 7489ce20df [Relay] ExprMutator Return Origin Expr When All Fields
Isn't Changed (#15237)
add 916542ed77 [TVMScript] Ensure completed root block has no read/write
(#15249)
add 5a78da4f3c [TIR] Output DeclBuffer in LowerTVMBuiltin (#15243)
add 24ae0d5b05 [bugfix][frontend][keras] Fix go_backwards attribute of
LSTM in keras frontend (#15261)
add 0c1aad78f9 [Testing] Add tvm.testing.local_run (#15268)
add a60cd0fecf [TIR] Allow symbolic bounds in IndexMap analysis (#15264)
new 2bb0ebbb8b Merge remote-tracking branch 'apache-upstream/main' into
unity-staging
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:
cmake/utils/FindVulkan.cmake | 57 ++--
docs/reference/api/links.rst | 1 +
include/tvm/relay/qnn/attrs.h | 2 +
include/tvm/tir/index_map.h | 12 +-
include/tvm/topi/transform.h | 9 +-
pyproject.toml | 1 -
python/tvm/contrib/emcc.py | 1 +
python/tvm/contrib/ndk.py | 55 ++++
.../tvm/relay/backend/contrib/ethosu/vela_api.py | 86 +++++-
python/tvm/relay/expr_functor.py | 56 +++-
python/tvm/relay/frontend/keras.py | 2 +
python/tvm/relay/qnn/op/qnn.py | 21 +-
python/tvm/te/schedule.py | 11 +-
python/tvm/testing/__init__.py | 2 +-
python/tvm/testing/{rpc_run.py => runner.py} | 85 +++++-
python/tvm/tir/analysis/analysis.py | 8 +-
python/tvm/tir/function.py | 20 +-
python/tvm/tir/op.py | 9 +-
python/tvm/tir/schedule/schedule.py | 40 ++-
python/tvm/tir/schedule/testing.py | 6 +-
src/arith/analyzer.cc | 32 +-
src/arith/ir_mutator_with_analyzer.cc | 20 ++
src/arith/ir_mutator_with_analyzer.h | 6 +
src/arith/iter_affine_map.cc | 6 +
src/arith/product_normal_form.h | 18 ++
src/relay/backend/te_compiler_cache.cc | 4 +-
src/relay/op/tensor/transform.cc | 4 +-
src/relay/qnn/op/dequantize.cc | 28 +-
src/relay/qnn/op/quantize.cc | 5 +-
src/relay/qnn/utils.h | 3 +-
src/runtime/logging.cc | 3 +
src/script/ir_builder/ir/ir.cc | 14 +-
src/target/source/codegen_cuda.cc | 3 +-
src/te/schedule/message_passing.cc | 14 +-
src/te/schedule/schedule_lang.cc | 6 +-
src/tir/analysis/verify_well_formed.cc | 25 +-
src/tir/ir/index_map.cc | 64 ++--
src/tir/ir/script/script_complete.cc | 20 +-
src/tir/schedule/analysis.h | 10 +
src/tir/schedule/analysis/analysis.cc | 14 +
src/tir/schedule/primitive.h | 4 +-
src/tir/schedule/primitive/cache_read_write.cc | 323 +++++++++++++++++----
src/tir/schedule/primitive/compute_at.cc | 14 -
.../schedule/primitive/layout_transformation.cc | 81 +++---
src/tir/schedule/transform.cc | 22 +-
src/tir/transforms/flatten_buffer.cc | 13 +-
src/tir/transforms/inject_double_buffer.cc | 4 +-
src/tir/transforms/ir_utils.cc | 3 +-
src/tir/transforms/lower_custom_datatypes.cc | 2 +-
src/tir/transforms/lower_thread_allreduce.cc | 2 +-
src/tir/transforms/lower_tvm_builtin.cc | 2 +
src/tir/transforms/lower_warp_memory.cc | 2 +-
src/tir/transforms/make_unpacked_api.cc | 11 +-
src/tir/transforms/storage_flatten.cc | 7 +-
src/tir/transforms/transform_mma_buffer_layout.cc | 6 +-
src/tir/transforms/update_pointer_storage_scope.cc | 9 +-
tests/python/contrib/test_ethosu/test_networks.py | 4 +-
.../contrib/test_ethosu/test_replace_conv2d.py | 14 +-
tests/python/contrib/test_ethosu/test_vela_api.py | 50 ++++
tests/python/frontend/keras/test_forward.py | 2 +
tests/python/relay/test_expr_functor.py | 2 +-
tests/python/relay/test_op_qnn_dequantize.py | 35 ++-
tests/python/relay/test_op_qnn_quantize.py | 23 ++
.../python/unittest/test_arith_iter_affine_map.py | 4 +-
...e_postproc_rewrite_parallel_vectorize_unroll.py | 2 -
.../test_meta_schedule_relay_integration.py | 20 +-
...meta_schedule_schedule_cuda_layout_transform.py | 6 +-
.../test_meta_schedule_schedule_rule_mlt_tc.py | 19 +-
.../unittest/test_meta_schedule_trace_apply.py | 10 +-
.../python/unittest/test_meta_schedule_tune_tir.py | 5 +
...sform_layout.py => test_te_transform_layout.py} | 0
.../test_tir_analysis_verify_well_formed.py | 1 +
.../{test_index_map.py => test_tir_index_map.py} | 11 +-
.../unittest/test_tir_schedule_cache_read_write.py | 156 ++++++++--
.../unittest/test_tir_schedule_transform_layout.py | 112 ++++++-
.../test_tir_transform_inject_ptx_async_copy.py | 97 ++++---
.../test_tir_transform_lower_tvm_builtin.py | 6 +-
...test_tir_transform_memhammer_lower_auto_copy.py | 8 -
.../python/unittest/test_tir_transform_simplify.py | 16 +-
tests/python/unittest/test_tvmscript_complete.py | 4 +
tests/python/unittest/test_tvmscript_roundtrip.py | 17 ++
tests/scripts/release/README.md | 2 +
tests/scripts/release/gather_prs.py | 22 +-
tests/scripts/release/make_notes.py | 12 +-
tests/scripts/task_python_docs.sh | 4 +-
85 files changed, 1509 insertions(+), 443 deletions(-)
rename python/tvm/testing/{rpc_run.py => runner.py} (66%)
rename tests/python/unittest/{test_transform_layout.py =>
test_te_transform_layout.py} (100%)
rename tests/python/unittest/{test_index_map.py => test_tir_index_map.py} (97%)