This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 3d5e440fff [ci] Disable flaky ethosu + roofline tests (#12956)
add 2d50979606 [TVMScript] Allow T.bool type annotations (#12975)
add 7fc35da3b9 [TEST] CPU feature detection for x86 and ARM dot product
instructions (#12980)
add 1b9e20a807 [ETHOSN] Transpose fully connected weights (#12970)
add 6147c31b36 [ETHOSN] Remove backwards copy in the runtime (#12968)
add 5e24aa2b78 [Hexagon] [runtime] Query for total and available VTCM
(#12992)
add 87c466cdee [Textures] Improve error reporting (#12986)
add e375c311da [Arith][IndexMap] Correct MapShape result for small
vectorized dims (#12927)
add 7ff42a1d4e [TIR] Require dtype.is_float() inside FloatImm (#12928)
add 50df4abf3b [Relay] optimize dumpir time (#12792)
add 78df9c1d79 [LOGGING] Optionally print backtrace on segfault (#12959)
add bb2cec1941 [Fix,MetaSchedule] Correct log usage in arithmetic
intensity feature (#12806)
add 6780c9f87d [MetaSchedule] Tuning API cleanup & ergonomics (#12895)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 1 +
cmake/config.cmake | 4 +
cmake/modules/LibInfo.cmake | 1 +
cmake/modules/Logging.cmake | 5 +
cmake/modules/contrib/CMSISNN.cmake | 4 +-
cmake/modules/contrib/EthosN.cmake | 3 +-
include/tvm/meta_schedule/measure_callback.h | 7 +-
include/tvm/meta_schedule/mutator.h | 15 +-
include/tvm/meta_schedule/postproc.h | 9 +
include/tvm/meta_schedule/schedule_rule.h | 18 +-
include/tvm/meta_schedule/search_strategy.h | 32 +-
include/tvm/meta_schedule/space_generator.h | 53 +-
include/tvm/meta_schedule/task_scheduler.h | 282 ++++-----
include/tvm/meta_schedule/tune_context.h | 87 +--
include/tvm/support/random_engine.h | 53 +-
python/tvm/contrib/torch/as_torch.py | 69 ++-
python/tvm/contrib/torch/optimize_torch.py | 179 +++---
python/tvm/meta_schedule/__init__.py | 19 +-
python/tvm/meta_schedule/builder/builder.py | 44 +-
python/tvm/meta_schedule/builder/local_builder.py | 6 +-
python/tvm/meta_schedule/cost_model/cost_model.py | 46 +-
python/tvm/meta_schedule/cost_model/mlp_model.py | 6 +-
.../tvm/meta_schedule/cost_model/random_model.py | 3 +-
python/tvm/meta_schedule/cost_model/xgb_model.py | 22 +-
python/tvm/meta_schedule/database/database.py | 87 +--
python/tvm/meta_schedule/default_config.py | 454 --------------
.../feature_extractor/feature_extractor.py | 26 +-
.../feature_extractor/random_feature_extractor.py | 8 +-
python/tvm/meta_schedule/logging.py | 259 ++++++++
.../tvm/meta_schedule/measure_callback/__init__.py | 7 +-
.../measure_callback/echo_statistics.py | 30 -
.../measure_callback/measure_callback.py | 16 +-
python/tvm/meta_schedule/mutator/mutator.py | 44 +-
python/tvm/meta_schedule/postproc/postproc.py | 36 +-
python/tvm/meta_schedule/profiler.py | 4 -
python/tvm/meta_schedule/relay_integration.py | 361 +++++++++--
python/tvm/meta_schedule/runner/local_runner.py | 4 +-
python/tvm/meta_schedule/runner/rpc_runner.py | 4 +-
python/tvm/meta_schedule/runner/runner.py | 37 +-
.../meta_schedule/schedule_rule/schedule_rule.py | 35 ++
.../search_strategy/evolutionary_search.py | 10 -
.../meta_schedule/search_strategy/replay_func.py | 11 +-
.../meta_schedule/search_strategy/replay_trace.py | 12 +-
.../search_strategy/search_strategy.py | 86 ++-
.../space_generator/post_order_apply.py | 26 +-
.../meta_schedule/space_generator/schedule_fn.py | 20 +-
.../space_generator/space_generator.py | 130 +++-
.../space_generator/space_generator_union.py | 20 +-
.../meta_schedule/task_scheduler/gradient_based.py | 49 +-
.../meta_schedule/task_scheduler/round_robin.py | 77 +--
.../meta_schedule/task_scheduler/task_scheduler.py | 282 ++++-----
.../meta_schedule/testing/dataset_extract_tasks.py | 4 +-
.../testing/dataset_sample_candidates.py | 17 +-
python/tvm/meta_schedule/testing/relay_workload.py | 52 +-
python/tvm/meta_schedule/testing/schedule_rule.py | 36 --
.../tvm/meta_schedule/testing/space_generation.py | 49 +-
python/tvm/meta_schedule/testing/tlcbench.py | 5 +-
python/tvm/meta_schedule/testing/torchbench/run.py | 135 ++--
python/tvm/meta_schedule/testing/tune_onnx.py | 52 +-
python/tvm/meta_schedule/testing/tune_relay.py | 44 +-
python/tvm/meta_schedule/testing/tune_te.py | 41 +-
python/tvm/meta_schedule/tir_integration.py | 155 +++++
python/tvm/meta_schedule/tune.py | 678 +++------------------
python/tvm/meta_schedule/tune_context.py | 201 +++---
python/tvm/meta_schedule/utils.py | 143 +----
python/tvm/script/tir/__init__.py | 1 +
python/tvm/script/tir/ty.py | 12 +-
python/tvm/testing/utils.py | 45 ++
python/tvm/tir/tensor_intrin/cuda.py | 7 +-
src/ir/expr.cc | 3 +
.../feature_extractor/per_store_feature.cc | 21 +-
.../measure_callback/add_to_database.cc | 6 +-
.../measure_callback/echo_statistics.cc | 134 ----
.../measure_callback/measure_callback.cc | 10 +
.../measure_callback/update_cost_model.cc | 11 +-
src/meta_schedule/mutator/mutator.cc | 30 +
src/meta_schedule/postproc/postproc.cc | 48 ++
.../postproc/rewrite_cooperative_fetch.cc | 2 +-
.../schedule_rule/cross_thread_reduction.cc | 4 +-
.../schedule_rule/multi_level_tiling.cc | 4 +-
.../schedule_rule/multi_level_tiling.h | 2 +-
.../multi_level_tiling_tensor_core.cc | 10 +-
.../multi_level_tiling_with_intrin.cc | 6 +-
src/meta_schedule/schedule_rule/schedule_rule.cc | 154 +++++
.../search_strategy/evolutionary_search.cc | 141 ++---
src/meta_schedule/search_strategy/replay_func.cc | 98 +--
src/meta_schedule/search_strategy/replay_trace.cc | 97 +--
.../search_strategy/search_strategy.cc | 5 +-
.../space_generator/post_order_apply.cc | 35 +-
src/meta_schedule/space_generator/schedule_fn.cc | 11 +-
.../space_generator/space_generator.cc | 96 +++
.../space_generator/space_generator_union.cc | 16 +-
src/meta_schedule/task_scheduler/gradient_based.cc | 218 ++-----
src/meta_schedule/task_scheduler/round_robin.cc | 23 +-
src/meta_schedule/task_scheduler/task_scheduler.cc | 405 ++++++++----
src/meta_schedule/tune_context.cc | 180 +-----
src/meta_schedule/utils.h | 57 +-
src/printer/relay_text_printer.cc | 9 +-
.../backend/contrib/cmsisnn/generate_constants.cc | 20 +-
src/relay/backend/contrib/constant_transforms.cc | 58 ++
src/relay/backend/contrib/constant_transforms.h | 59 ++
src/relay/backend/contrib/ethosn/codegen.cc | 4 +-
.../backend/contrib/ethosn/convert_equivalent.cc | 1 +
src/relay/backend/contrib/ethosn/ethosn_api.cc | 49 +-
src/relay/backend/contrib/ethosn/ethosn_api.h | 13 +-
src/relay/backend/te_compiler.cc | 2 +-
src/relay/backend/te_compiler_cache.cc | 14 +-
src/relay/backend/utils.h | 7 +
src/runtime/contrib/ethosn/ethosn_device.cc | 64 +-
src/runtime/hexagon/hexagon_vtcm_pool.cc | 19 +-
src/runtime/hexagon/hexagon_vtcm_pool.h | 4 +-
src/runtime/logging.cc | 29 +
src/support/libinfo.cc | 1 +
src/tir/ir/index_map.cc | 45 +-
src/tir/schedule/concrete_schedule.cc | 2 +-
src/tir/transforms/texture_flatten.cc | 3 +-
.../contrib/test_ethosn/test_fullyconnected.py | 56 +-
tests/python/contrib/test_ethosn/test_networks.py | 4 +-
.../contrib/test_hexagon/test_meta_schedule.py | 115 ++--
...le_auto_tensorize.py => test_auto_tensorize.py} | 198 +++---
.../{test_tuning.py => test_legacy_tuning.py} | 0
tests/python/integration/test_tuning.py | 433 +++----------
tests/python/relay/test_op_level1.py | 2 +-
tests/python/relay/test_op_level10.py | 2 +-
tests/python/relay/test_op_level2.py | 65 +-
tests/python/unittest/test_index_map.py | 9 +-
.../unittest/test_meta_schedule_cost_model.py | 17 +-
.../test_meta_schedule_custom_rule_winograd_cpu.py | 4 -
...test_meta_schedule_custom_rule_winograd_cuda.py | 3 -
...schedule_feature_extractor_per_store_feature.py | 46 +-
.../test_meta_schedule_measure_callback.py | 18 +-
...eta_schedule_mutator_mutate_compute_location.py | 17 +-
.../test_meta_schedule_mutator_mutate_parallel.py | 17 +-
..._meta_schedule_mutator_mutate_thread_binding.py | 17 +-
.../test_meta_schedule_mutator_mutate_tile_size.py | 15 +-
.../test_meta_schedule_mutator_mutate_unroll.py | 17 +-
.../test_meta_schedule_post_order_apply.py | 50 +-
...meta_schedule_postproc_disallow_dynamic_loop.py | 21 +-
..._schedule_postproc_rewrite_cooperative_fetch.py | 22 +-
.../test_meta_schedule_postproc_rewrite_layout.py | 83 ++-
...ta_schedule_postproc_rewrite_reduction_block.py | 21 +-
...est_meta_schedule_postproc_rewrite_tensorize.py | 26 +-
...meta_schedule_postproc_rewrite_unbound_block.py | 26 +-
.../test_meta_schedule_postproc_verify_gpu_code.py | 23 +-
....py => test_meta_schedule_relay_integration.py} | 208 ++++++-
...test_meta_schedule_schedule_rule_add_rfactor.py | 23 +-
.../test_meta_schedule_schedule_rule_auto_bind.py | 33 +-
...test_meta_schedule_schedule_rule_auto_inline.py | 43 +-
...chedule_schedule_rule_cross_thread_reduction.py | 51 +-
.../test_meta_schedule_schedule_rule_mlt.py | 61 +-
.../test_meta_schedule_schedule_rule_mlt_intrin.py | 19 +-
.../test_meta_schedule_schedule_rule_mlt_tc.py | 80 ++-
...dule_schedule_rule_parallel_vectorize_unroll.py | 21 +-
...hedule_schedule_rule_random_compute_location.py | 13 +-
.../unittest/test_meta_schedule_search_strategy.py | 45 +-
.../unittest/test_meta_schedule_space_cpu.py | 106 +---
.../unittest/test_meta_schedule_space_cuda.py | 116 +---
.../unittest/test_meta_schedule_space_generator.py | 7 +-
.../unittest/test_meta_schedule_task_scheduler.py | 155 ++---
.../unittest/test_meta_schedule_tune_relay.py | 553 -----------------
.../python/unittest/test_meta_schedule_tune_te.py | 52 --
.../python/unittest/test_meta_schedule_tune_tir.py | 115 ++--
.../test_meta_schedule_vnni_integration.py | 249 ++++++++
tests/python/unittest/test_tir_constructor.py | 11 +-
tests/python/unittest/test_tir_schedule_trace.py | 1 -
tests/python/unittest/test_tvmscript_roundtrip.py | 46 ++
tests/scripts/task_config_build_arm.sh | 1 +
tests/scripts/task_config_build_cortexm.sh | 1 +
tests/scripts/task_config_build_cpu.sh | 1 +
tests/scripts/task_config_build_gpu.sh | 1 +
tests/scripts/task_config_build_gpu_other.sh | 1 +
tests/scripts/task_config_build_hexagon.sh | 1 +
tests/scripts/task_config_build_i386.sh | 1 +
173 files changed, 5003 insertions(+), 5349 deletions(-)
delete mode 100644 python/tvm/meta_schedule/default_config.py
create mode 100644 python/tvm/meta_schedule/logging.py
delete mode 100644 python/tvm/meta_schedule/measure_callback/echo_statistics.py
delete mode 100644 python/tvm/meta_schedule/testing/schedule_rule.py
create mode 100644 python/tvm/meta_schedule/tir_integration.py
delete mode 100644 src/meta_schedule/measure_callback/echo_statistics.cc
create mode 100644 src/relay/backend/contrib/constant_transforms.cc
create mode 100644 src/relay/backend/contrib/constant_transforms.h
rename tests/python/integration/{test_meta_schedule_auto_tensorize.py =>
test_auto_tensorize.py} (72%)
copy tests/python/integration/{test_tuning.py => test_legacy_tuning.py} (100%)
rename tests/python/unittest/{test_meta_schedule_integration.py =>
test_meta_schedule_relay_integration.py} (55%)
delete mode 100644 tests/python/unittest/test_meta_schedule_tune_relay.py
delete mode 100644 tests/python/unittest/test_meta_schedule_tune_te.py
create mode 100644 tests/python/unittest/test_meta_schedule_vnni_integration.py
