This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 58daeb4905 Update FlashInfer (#16292)
add 303afdbccc [Unity][MSC][M3.2] Add gym for pruning and quantization,
enable auto prune/quantize (#16280)
No new revisions were added by this update.
Summary of changes:
.../contrib/{ethosu => msc/core/gym}/__init__.py | 6 +-
.../core/{tools/prune => gym/agent}/__init__.py | 4 +-
.../tvm/contrib/msc/core/gym/agent/base_agent.py | 312 ++++++++
python/tvm/contrib/msc/core/gym/agent/method.py | 80 ++
.../tvm/contrib/msc/core/gym/agent/search_agent.py | 180 +++++
.../{ethosu => msc/core/gym/control}/__init__.py | 5 +-
.../tvm/contrib/msc/core/gym/control/configer.py | 97 +++
.../tvm/contrib/msc/core/gym/control/controller.py | 107 +++
.../__init__.py => gym/control/namespace.py} | 25 +-
python/tvm/contrib/msc/core/gym/control/service.py | 810 +++++++++++++++++++++
python/tvm/contrib/msc/core/gym/control/worker.py | 216 ++++++
.../{tools/distill => gym/environment}/__init__.py | 5 +-
.../contrib/msc/core/gym/environment/base_env.py | 366 ++++++++++
.../tvm/contrib/msc/core/gym/environment/method.py | 202 +++++
.../contrib/msc/core/gym/environment/prune_env.py | 95 +++
.../msc/core/gym/environment/quantize_env.py | 99 +++
python/tvm/contrib/msc/core/tools/prune/pruner.py | 54 ++
python/tvm/contrib/msc/core/tools/tool.py | 16 +
python/tvm/contrib/msc/core/utils/file.py | 1 +
python/tvm/contrib/msc/core/utils/register.py | 172 +++++
.../msc/framework/tensorrt/codegen/codegen.py | 19 +-
python/tvm/contrib/msc/pipeline/manager.py | 22 +-
tests/python/contrib/test_msc/test_tools.py | 63 +-
23 files changed, 2934 insertions(+), 22 deletions(-)
copy python/tvm/contrib/{ethosu => msc/core/gym}/__init__.py (88%)
copy python/tvm/contrib/msc/core/{tools/prune => gym/agent}/__init__.py (92%)
create mode 100644 python/tvm/contrib/msc/core/gym/agent/base_agent.py
create mode 100644 python/tvm/contrib/msc/core/gym/agent/method.py
create mode 100644 python/tvm/contrib/msc/core/gym/agent/search_agent.py
copy python/tvm/contrib/{ethosu => msc/core/gym/control}/__init__.py (89%)
create mode 100644 python/tvm/contrib/msc/core/gym/control/configer.py
create mode 100644 python/tvm/contrib/msc/core/gym/control/controller.py
copy python/tvm/contrib/msc/core/{runtime/__init__.py =>
gym/control/namespace.py} (65%)
create mode 100644 python/tvm/contrib/msc/core/gym/control/service.py
create mode 100644 python/tvm/contrib/msc/core/gym/control/worker.py
copy python/tvm/contrib/msc/core/{tools/distill =>
gym/environment}/__init__.py (89%)
create mode 100644 python/tvm/contrib/msc/core/gym/environment/base_env.py
create mode 100644 python/tvm/contrib/msc/core/gym/environment/method.py
create mode 100644 python/tvm/contrib/msc/core/gym/environment/prune_env.py
create mode 100644 python/tvm/contrib/msc/core/gym/environment/quantize_env.py
