This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch ir-builder
in repository https://gitbox.apache.org/repos/asf/tvm.git
discard e36f448b62 Squashed
add fc411dc6fa Pass that removes reshapes post LowerTE (#12215)
add 9b86009553 [OpenCL] Use size_t instead of int64_t for OpenCL timer
count (#12328)
add b4e36c61fc [CI] Increase the number of shards for Cortex-M from 4 to
8. (#12334)
add 8e133b1990 [CI][Docker] Removes Dockerfile.ci_qemu as it was moved to
Dockerfile.ci_cortexm (#12329)
add 6ebb0fd1ad [ETHOSN] Fix output tensor ordering (#12317)
add b79f9501fd [Relay][Op] Multinomial (#12284)
add 7f800e41de [CI] Deduplicate and clean XML test reports (#12332)
add 6026a5e185 [ci] Fix aws s3 cp command in the Jenkinsfile (#12341)
add ef39e46a1d [microTVM][Zephyr] Fix missing BOARD in CMakeCache file
(#12338)
add d6be6940bd [BYOC-DNNL] Bug Fix (#12314)
add bd763d3c23 [Topi] add x86 schedule for batch_norm (#12321)
add 151d6ab8ac [FIX,ROOFLINE] Only save tir functions for roofline (#12339)
add 5fe756a965 [MetaSchedule][Bugfix] Feature: Strides of buffer access
(#12331)
add 6d87828806 [TIR] Simplify indices in InjectVirtualThread (#12259)
add d4dff5dfd6 [COMMUNITY] Yuanjing Shi -> Reviewer (#12345)
add 0c3c08372a [hexagon][topi] add sliced max_pool2 (#12169)
add 4c4fcee4c2 [UMA] UMA v1.0 (#12087)
add 52d6b59a39 [AutoTVM][Fix] Fix wrong axis names of data_vec (#12303)
add 23b203ed3d [Hexagon] Minor changes/fixes in codegen_hexagon.cc (#12308)
add 0aab92b40b [Node] fix typos in include/tvm/node/functor.h
add aea82c6417 [MetaSchedule] Extend tune_tir to support tuning of
specific blocks. (#12342)
add 811584992c Infer the value of shape expr to avoid dynamic (#12313)
add 0c281b7064 [FIX,STORAGE REWRITE] Rewrite buffers in let statements
(#12349)
add a6c3b1f6c3 [TIR] Add int8 CUDA tensor core intrinsics (#12354)
add 1a98ea2e4f [TIR] Minor fix to tensor intrin description (#12356)
new 66c2087bd0 Squashed
This update added new revisions after undoing existing revisions.
That is to say, some revisions that were in the old version of the
branch are not in the new version. This situation occurs
when a user --force pushes a change and generates a repository
containing something like this:
* -- * -- B -- O -- O -- O (e36f448b62)
\
N -- N -- N refs/heads/ir-builder (66c2087bd0)
You should already have received notification emails for all of the O
revisions, and so the following emails describe only the N revisions
from the common base, B.
Any revisions marked "omit" are not gone; other references still
refer to them. Any revisions marked "discard" are gone forever.
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 | 2 +
CONTRIBUTORS.md | 1 +
Jenkinsfile | 438 +++++++++++++++++----
.../zephyr/template_project/microtvm_api_server.py | 35 +-
.../vta/exec => apps/uma/_template}/__init__.py | 6 +-
apps/uma/_template/backend.py | 45 +++
.../conftest.py => apps/uma/_template/codegen.py | 17 +-
apps/uma/_template/conv2dnchw.cc | 96 +++++
apps/uma/_template/passes.py | 136 +++++++
.../read_results.py => uma/_template/patterns.py} | 17 +-
apps/uma/_template/run.py | 82 ++++
.../_template/strategies.py} | 37 +-
apps/uma/uma_cli.py | 98 +++++
ci/jenkins/Test.groovy.j2 | 2 +-
ci/jenkins/macros.j2 | 10 +-
cmake/config.cmake | 3 +
cmake/modules/LibInfo.cmake | 1 +
.../modules/contrib/{CODEGENC.cmake => UMA.cmake} | 8 +-
conftest.py | 11 +-
docker/Dockerfile.ci_qemu | 120 ------
docs/conf.py | 1 +
gallery/tutorial/uma.py | 292 ++++++++++++++
include/tvm/meta_schedule/space_generator.h | 2 +-
include/tvm/node/functor.h | 4 +-
include/tvm/relay/attrs/random.h | 10 +
include/tvm/relay/transform.h | 11 +
include/tvm/tir/ir_builder.h | 2 +
python/tvm/meta_schedule/default_config.py | 5 +-
.../space_generator/post_order_apply.py | 12 +-
python/tvm/meta_schedule/tune.py | 27 ++
.../backend/contrib/uma}/__init__.py | 8 +-
.../tvm/relay/backend/contrib/uma/api/__init__.py | 14 +-
.../contrib/{ethosu => uma/api}/_ffi_api.py | 3 +-
.../tvm/relay/backend/contrib/uma/api/codegen.py | 64 +++
python/tvm/relay/backend/contrib/uma/api/lower.py | 165 ++++++++
.../relay/backend/contrib/uma/api/partitioner.py | 122 ++++++
python/tvm/relay/backend/contrib/uma/api/utils.py | 73 ++++
python/tvm/relay/backend/contrib/uma/backend.py | 293 ++++++++++++++
python/tvm/relay/frontend/onnx.py | 22 ++
python/tvm/relay/frontend/pytorch.py | 16 +
python/tvm/relay/frontend/tflite.py | 18 +-
python/tvm/relay/op/contrib/dnnl.py | 48 ++-
python/tvm/relay/op/random/_kernel.py | 2 +
python/tvm/relay/op/random/kernel.py | 41 ++
python/tvm/relay/op/strategy/generic.py | 22 ++
python/tvm/relay/op/strategy/x86.py | 12 +
python/tvm/script/parser/tir/parser.py | 4 +-
python/tvm/script/parser/utils.py | 17 +-
python/tvm/script/parser_v1/tir/__init__.py | 12 +-
python/tvm/script/parser_v1/tir/intrin.py | 84 +---
python/tvm/script/parser_v1/tir/ty.py | 15 +-
python/tvm/te/operation.py | 15 +-
python/tvm/testing/aot.py | 12 +-
python/tvm/tir/__init__.py | 3 +
python/tvm/tir/function.py | 2 +-
python/tvm/tir/ir_builder_v2.py | 30 ++
python/tvm/tir/op.py | 12 +
python/tvm/tir/tensor_intrin/arm_cpu.py | 2 -
python/tvm/tir/tensor_intrin/cuda.py | 79 +++-
python/tvm/tir/tensor_intrin/rocm.py | 2 +-
python/tvm/tir/tensor_intrin/x86.py | 2 -
python/tvm/topi/generic/conv2d.py | 4 +-
python/tvm/topi/hexagon/slice_ops/__init__.py | 1 +
python/tvm/topi/hexagon/slice_ops/max_pool2d.py | 196 +++++++++
python/tvm/topi/random/kernel.py | 56 +++
python/tvm/topi/scatter.py | 2 +-
python/tvm/topi/x86/nn.py | 30 ++
python/tvm/utils/roofline/__init__.py | 13 +-
.../feature_extractor/per_store_feature.cc | 4 +-
.../space_generator/post_order_apply.cc | 30 +-
src/relay/backend/aot_executor_codegen.cc | 7 +-
src/relay/backend/contrib/uma/relay_to_tir.cc | 175 ++++++++
src/relay/backend/contrib/uma/targets.cc | 80 ++++
.../tir_to_runtime.cc | 51 ++-
src/relay/op/random/kernel.cc | 46 +++
src/relay/transforms/remove_standalone_reshapes.cc | 120 ++++++
src/runtime/contrib/ethosn/ethosn_device.cc | 6 +-
src/runtime/opencl/opencl_common.h | 8 +-
src/runtime/opencl/opencl_device_api.cc | 4 +-
src/support/libinfo.cc | 1 +
src/target/llvm/codegen_hexagon.cc | 78 ++--
src/tir/ir_builder/ir_builder.cc | 2 +
src/tir/schedule/primitive/cache_read_write.cc | 2 +-
src/tir/transforms/inject_virtual_thread.cc | 28 +-
src/tir/transforms/storage_rewrite.cc | 8 +
tests/python/contrib/test_cmsisnn/test_conv2d.py | 1 +
tests/python/contrib/test_cmsisnn/test_pooling.py | 8 +-
.../contrib/test_cmsisnn/test_remove_reshapes.py | 169 ++++++++
.../python/contrib/test_ethosn/test_topologies.py | 46 +++
tests/python/contrib/test_ethosu/test_networks.py | 10 +-
...vg_pool2d_slice.py => test_max_pool2d_slice.py} | 53 +--
tests/python/contrib/test_uma/test_partition.py | 97 +++++
tests/python/contrib/test_uma/test_target.py | 85 ++++
.../test_uma/test_uma_lowering_with_umalower.py | 121 ++++++
tests/python/contrib/test_uma/test_uma_pipeline.py | 136 +++++++
tests/python/contrib/test_uma/test_uma_utils.py | 87 ++++
.../test_uma/test_uma_vanilla_accelerator.py | 56 +++
tests/python/frontend/onnx/test_forward.py | 46 ++-
tests/python/frontend/pytorch/test_forward.py | 50 ++-
tests/python/relay/aot/test_crt_aot.py | 2 +-
tests/python/relay/aot/test_crt_aot_usmp.py | 36 +-
.../test_pass_remove_standalone_reshapes.py | 260 ++++++++++++
tests/python/relay/test_pass_partition_graph.py | 42 +-
tests/python/topi/python/test_topi_batch_norm.py | 1 +
tests/python/topi/python/test_topi_prng.py | 92 ++++-
.../test_meta_schedule_post_order_apply.py | 82 ++--
.../unittest/test_meta_schedule_space_cuda.py | 2 +-
.../python/unittest/test_meta_schedule_tune_tir.py | 57 ++-
.../test_tir_transform_inject_virtual_thread.py | 63 ++-
.../unittest/test_tir_transform_storage_rewrite.py | 24 ++
tests/scripts/setup-pytest-env.sh | 14 +-
tests/scripts/task_config_build_arm.sh | 1 +
tests/scripts/task_config_build_cortexm.sh | 2 +
tests/scripts/task_config_build_cpu.sh | 1 +
tests/scripts/task_config_build_i386.sh | 1 +
tests/scripts/task_config_build_qemu.sh | 35 --
116 files changed, 4660 insertions(+), 717 deletions(-)
copy {vta/python/vta/exec => apps/uma/_template}/__init__.py (94%)
create mode 100644 apps/uma/_template/backend.py
copy tests/python/contrib/test_opencl/conftest.py =>
apps/uma/_template/codegen.py (77%)
create mode 100644 apps/uma/_template/conv2dnchw.cc
create mode 100644 apps/uma/_template/passes.py
copy apps/{sgx/read_results.py => uma/_template/patterns.py} (74%)
create mode 100644 apps/uma/_template/run.py
copy apps/{microtvm/reference-vm/base-box/base_box_test.sh =>
uma/_template/strategies.py} (54%)
mode change 100755 => 100644
create mode 100644 apps/uma/uma_cli.py
copy cmake/modules/contrib/{CODEGENC.cmake => UMA.cmake} (82%)
delete mode 100644 docker/Dockerfile.ci_qemu
create mode 100644 gallery/tutorial/uma.py
copy python/tvm/{topi/cpp/vision => relay/backend/contrib/uma}/__init__.py
(84%)
copy tests/python/contrib/test_opencl/conftest.py =>
python/tvm/relay/backend/contrib/uma/api/__init__.py (78%)
copy python/tvm/relay/backend/contrib/{ethosu => uma/api}/_ffi_api.py (89%)
create mode 100644 python/tvm/relay/backend/contrib/uma/api/codegen.py
create mode 100644 python/tvm/relay/backend/contrib/uma/api/lower.py
create mode 100644 python/tvm/relay/backend/contrib/uma/api/partitioner.py
create mode 100644 python/tvm/relay/backend/contrib/uma/api/utils.py
create mode 100644 python/tvm/relay/backend/contrib/uma/backend.py
create mode 100644 python/tvm/topi/hexagon/slice_ops/max_pool2d.py
create mode 100644 src/relay/backend/contrib/uma/relay_to_tir.cc
create mode 100644 src/relay/backend/contrib/uma/targets.cc
copy src/relay/backend/contrib/{example_target_hooks => uma}/tir_to_runtime.cc
(58%)
create mode 100644 src/relay/transforms/remove_standalone_reshapes.cc
create mode 100644 tests/python/contrib/test_cmsisnn/test_remove_reshapes.py
copy tests/python/contrib/test_hexagon/topi/{test_avg_pool2d_slice.py =>
test_max_pool2d_slice.py} (92%)
create mode 100644 tests/python/contrib/test_uma/test_partition.py
create mode 100644 tests/python/contrib/test_uma/test_target.py
create mode 100644
tests/python/contrib/test_uma/test_uma_lowering_with_umalower.py
create mode 100644 tests/python/contrib/test_uma/test_uma_pipeline.py
create mode 100644 tests/python/contrib/test_uma/test_uma_utils.py
create mode 100644
tests/python/contrib/test_uma/test_uma_vanilla_accelerator.py
create mode 100644
tests/python/relay/backend/test_pass_remove_standalone_reshapes.py
delete mode 100755 tests/scripts/task_config_build_qemu.sh
