This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 918fc4ecf7 [Unity][Dlight] Matmul Rules (#15191)
add b4b27fafbd [Python] Enhance Wheel Packaging (#15167)
add e1013f10f4 [CMake] Support LLVM-16 static linking (#15164)
add 34637d7ee3 [TensorIR][Visitor] Visit buffer members in
`match_buffer`'s in block visitor functions (#15153)
add e178375e1b [Frontend][Relay][Keras] Fix concatenate convert function
in axis parsing (#15175)
add 977b4b2e05 Update tvm_runtime.h (#15185)
add 683dfb0c04 [RPC] Report RPC Session Timeout to Client Instead of
"kShutdown" (#15187)
add fb64be3f78 [ARITH] Allow Analyzer to MarkGlobalNonNegValue (#15193)
add 5931cf10eb [TIR][Transform] Add LiftThreadBinding Pass (#15207)
add f14c61f0d1 [AOT] Remove workaround to help resolve test flakiness
(#15181)
add 03ef29e8ad [TIR][Schedule] Derive Nonnegative Bounds from Shape Var
(#15210)
add b160cb1a10 Update version to 0.14.dev0 on main branch (#15208)
add a60b815fe5 [TIR] Support cross-threaad reduction lowering with
thread-broadcasting rewrite (#15192)
new 40300a3b92 [MERGE] Merge main into unity 2023-07-03
new 1631c3d4ce [MERGE] Fix testcase after merge
The 2 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:
.github/workflows/main.yml | 21 ---
apps/android_rpc/app/src/main/jni/tvm_runtime.h | 1 +
cmake/utils/FindLLVM.cmake | 32 +++-
conda/recipe/meta.yaml | 2 +-
include/tvm/arith/analyzer.h | 16 ++
include/tvm/runtime/c_runtime_api.h | 2 +-
include/tvm/tir/transform.h | 6 +
include/tvm/tir/var.h | 6 +
python/gen_requirements.py | 6 +-
python/setup.py | 39 ++++-
python/tvm/_ffi/libinfo.py | 2 +-
python/tvm/error.py | 5 +
python/tvm/relay/frontend/keras.py | 11 +-
python/tvm/rpc/server.py | 88 +++++-----
python/tvm/testing/aot.py | 10 +-
python/tvm/tir/transform/transform.py | 11 ++
src/arith/analyzer.cc | 33 ++++
src/arith/const_int_bound.cc | 38 +++-
src/arith/ir_mutator_with_analyzer.cc | 9 +
src/arith/ir_mutator_with_analyzer.h | 8 +
src/driver/driver_api.cc | 1 +
src/meta_schedule/postproc/verify_gpu_code.cc | 1 +
src/runtime/rpc/rpc_endpoint.cc | 8 +-
src/runtime/rpc/rpc_socket_impl.cc | 34 ++++
src/tir/ir/expr.cc | 8 +-
src/tir/ir/stmt_functor.cc | 32 +++-
src/tir/schedule/primitive/compute_at.cc | 15 ++
src/tir/transforms/flatten_buffer.cc | 1 +
src/tir/transforms/lift_thread_binding.cc | 195 +++++++++++++++++++++
src/tir/transforms/lower_cross_thread_reduction.cc | 158 +++++++++++++++--
src/tir/transforms/simplify.cc | 21 +--
tests/python/dlight/test_gpu_reduction.py | 4 +-
tests/python/frontend/keras/test_forward.py | 19 ++
.../python/unittest/test_arith_const_int_bound.py | 17 ++
tests/python/unittest/test_runtime_rpc.py | 31 ++++
.../python/unittest/test_tir_lower_match_buffer.py | 11 +-
.../unittest/test_tir_schedule_compute_at.py | 69 ++++++++
.../test_tir_transform_lift_thread_binding.py | 139 +++++++++++++++
...t_tir_transform_lower_cross_thread_reduction.py | 174 ++++++++++++++++++
.../python/unittest/test_tir_transform_simplify.py | 16 ++
.../test_tir_transform_unify_thread_binding.py | 43 +++++
.../scripts/task_config_build_minimal_cross_isa.sh | 1 +
version.py | 2 +-
web/package.json | 2 +-
44 files changed, 1196 insertions(+), 152 deletions(-)
create mode 100644 src/tir/transforms/lift_thread_binding.cc
create mode 100644
tests/python/unittest/test_tir_transform_lift_thread_binding.py
