This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch ci-gpu-torch-1.11
in repository https://gitbox.apache.org/repos/asf/tvm.git.
omit 9c78a55 fixed keras tutorial
omit c725954 fix onnx test
omit de7c4e8 Validating the new gpu image
add 5629f8a [MetaSchedule] Add Gradient Based Task Scheduler (#10366)
add 8226bd0 [microNPU] Some housekeeping in the test_ethosu folder
(#10824)
add 8775a80 [TVMC] Support compiling and running with VM (#10722)
new 6d92d26 Update Jenkinsfile to point to the new GPU image
new 87b7051 fix onnx test
new ecd4ba2 fixed keras tutorial
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 (9c78a55)
\
N -- N -- N refs/heads/ci-gpu-torch-1.11 (ecd4ba2)
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 3 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:
Jenkinsfile | 2 +-
include/tvm/meta_schedule/search_strategy.h | 12 +-
include/tvm/meta_schedule/task_scheduler.h | 94 ++---
include/tvm/meta_schedule/tune_context.h | 4 +-
include/tvm/support/random_engine.h | 18 +-
include/tvm/tir/schedule/schedule.h | 2 +-
jenkins/Jenkinsfile.j2 | 2 +-
python/tvm/driver/tvmc/compiler.py | 77 +++-
python/tvm/driver/tvmc/model.py | 94 ++++-
python/tvm/driver/tvmc/runner.py | 127 ++++---
.../search_strategy/evolutionary_search.py | 13 +-
.../meta_schedule/search_strategy/replay_func.py | 13 +-
.../meta_schedule/search_strategy/replay_trace.py | 16 +-
.../tvm/meta_schedule/task_scheduler/__init__.py | 1 +
.../{round_robin.py => gradient_based.py} | 61 ++--
.../meta_schedule/task_scheduler/round_robin.py | 15 +-
.../meta_schedule/task_scheduler/task_scheduler.py | 86 ++---
.../testing/tune_relay_meta_schedule.py | 8 +-
.../meta_schedule/testing/tune_te_meta_schedule.py | 3 +-
python/tvm/meta_schedule/tune.py | 14 +-
python/tvm/meta_schedule/utils.py | 2 +-
.../measure_callback/echo_statistics.cc | 10 +-
.../search_strategy/evolutionary_search.cc | 16 +-
src/meta_schedule/search_strategy/replay_func.cc | 12 +-
src/meta_schedule/search_strategy/replay_trace.cc | 12 +-
src/meta_schedule/task_scheduler/gradient_based.cc | 228 ++++++++++++
src/meta_schedule/task_scheduler/round_robin.cc | 10 +-
src/meta_schedule/task_scheduler/task_scheduler.cc | 134 ++++---
src/meta_schedule/tune_context.cc | 5 +-
src/meta_schedule/utils.h | 23 ++
src/support/table_printer.h | 154 ++++++++
src/tir/schedule/concrete_schedule.cc | 3 -
src/tir/schedule/concrete_schedule.h | 2 +-
tests/python/contrib/test_ethosu/infra.py | 141 ++++++--
tests/python/contrib/test_ethosu/test_codegen.py | 393 +++++----------------
.../contrib/test_ethosu/test_identity_optimizer.py | 7 +-
tests/python/contrib/test_ethosu/test_legalize.py | 14 -
.../contrib/test_ethosu/test_lut_optimizer.py | 3 +-
tests/python/driver/tvmc/test_compiler.py | 50 +--
tests/python/driver/tvmc/test_model.py | 21 +-
tests/python/driver/tvmc/test_runner.py | 8 +-
.../test_meta_schedule_measure_callback.py | 9 +-
.../unittest/test_meta_schedule_search_strategy.py | 16 +-
.../unittest/test_meta_schedule_task_scheduler.py | 138 ++++++--
.../unittest/test_meta_schedule_tune_relay.py | 23 +-
.../python/unittest/test_meta_schedule_tune_te.py | 3 +-
.../python/unittest/test_meta_schedule_tune_tir.py | 25 +-
47 files changed, 1310 insertions(+), 814 deletions(-)
copy python/tvm/meta_schedule/task_scheduler/{round_robin.py =>
gradient_based.py} (62%)
create mode 100644 src/meta_schedule/task_scheduler/gradient_based.cc
create mode 100644 src/support/table_printer.h
