This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 831d769f47 [Transform] Remove R.Object parameters after
LazyTransformParams (#16699)
add 3b25697e9d [MSC][M5.2] Enable quantize && prune with gym by wrapper
(#16702)
No new revisions were added by this update.
Summary of changes:
gallery/how_to/work_with_msc/using_tools.py | 11 +-
.../tvm/contrib/msc/core/gym/agent/base_agent.py | 49 +++---
python/tvm/contrib/msc/core/gym/agent/method.py | 11 +-
.../tvm/contrib/msc/core/gym/agent/search_agent.py | 11 +-
.../tvm/contrib/msc/core/gym/control/configer.py | 12 +-
.../tvm/contrib/msc/core/gym/control/controller.py | 6 +-
python/tvm/contrib/msc/core/gym/control/service.py | 46 ++++--
python/tvm/contrib/msc/core/gym/control/worker.py | 32 ++--
.../contrib/msc/core/gym/environment/base_env.py | 105 +++++++++---
.../tvm/contrib/msc/core/gym/environment/method.py | 14 +-
.../contrib/msc/core/gym/environment/prune_env.py | 51 +++---
.../msc/core/gym/environment/quantize_env.py | 58 ++-----
.../msc/core/gym/{control => }/namespace.py | 2 +-
python/tvm/contrib/msc/core/runtime/hook.py | 4 +-
python/tvm/contrib/msc/core/runtime/runner.py | 43 ++++-
python/tvm/contrib/msc/core/tools/configer.py | 20 +--
.../contrib/msc/core/tools/distill/distiller.py | 4 +-
.../tvm/contrib/msc/core/tools/distill/method.py | 6 +-
python/tvm/contrib/msc/core/tools/execute.py | 2 +-
python/tvm/contrib/msc/core/tools/prune/method.py | 6 +-
python/tvm/contrib/msc/core/tools/prune/pruner.py | 4 +-
.../tvm/contrib/msc/core/tools/quantize/method.py | 6 +-
.../contrib/msc/core/tools/quantize/quantizer.py | 4 +-
.../tvm/contrib/msc/core/tools/track/configer.py | 15 +-
python/tvm/contrib/msc/core/tools/track/method.py | 4 +-
python/tvm/contrib/msc/core/tools/track/tracker.py | 4 +-
python/tvm/contrib/msc/core/utils/expr.py | 25 ++-
python/tvm/contrib/msc/core/utils/file.py | 26 ++-
python/tvm/contrib/msc/core/utils/log.py | 8 +
python/tvm/contrib/msc/core/utils/message.py | 2 +
python/tvm/contrib/msc/core/utils/register.py | 181 +++++++++------------
.../tensorflow/tools/distill/distiller.py | 5 +-
.../msc/framework/tensorflow/tools/prune/pruner.py | 5 +-
.../tensorflow/tools/quantize/quantizer.py | 5 +-
.../framework/tensorflow/tools/track/tracker.py | 5 +-
.../msc/framework/tensorrt/runtime/runner.py | 23 +++
.../framework/tensorrt/tools/distill/distiller.py | 5 +-
.../msc/framework/tensorrt/tools/prune/pruner.py | 5 +-
.../framework/tensorrt/tools/quantize/method.py | 4 +-
.../framework/tensorrt/tools/quantize/quantizer.py | 5 +-
.../msc/framework/tensorrt/tools/track/tracker.py | 5 +-
.../msc/framework/torch/tools/distill/distiller.py | 5 +-
.../msc/framework/torch/tools/distill/method.py | 4 +-
.../msc/framework/torch/tools/prune/pruner.py | 5 +-
.../msc/framework/torch/tools/quantize/method.py | 4 +-
.../framework/torch/tools/quantize/quantizer.py | 5 +-
.../msc/framework/torch/tools/track/tracker.py | 5 +-
.../contrib/msc/framework/tvm/runtime/runner.py | 38 ++++-
.../msc/framework/tvm/tools/distill/distiller.py | 5 +-
.../msc/framework/tvm/tools/prune/pruner.py | 5 +-
.../msc/framework/tvm/tools/quantize/method.py | 4 +-
.../msc/framework/tvm/tools/quantize/quantizer.py | 5 +-
.../msc/framework/tvm/tools/track/tracker.py | 5 +-
python/tvm/contrib/msc/pipeline/config.py | 14 +-
python/tvm/contrib/msc/pipeline/manager.py | 140 +++++++++++-----
python/tvm/contrib/msc/pipeline/wrapper.py | 26 +--
56 files changed, 688 insertions(+), 421 deletions(-)
copy python/tvm/contrib/msc/core/gym/{control => }/namespace.py (96%)
