1 file changed
+1
-1
lines changed- cmake/modules/LibInfo.cmake+1
- gallery/how_to/work_with_msc/using_tools.py+9-2
- include/tvm/relax/attrs/nn.h+44
- python/tvm/contrib/msc/core/gym/agent/base_agent.py+30-19
- python/tvm/contrib/msc/core/gym/agent/method.py+7-4
- python/tvm/contrib/msc/core/gym/agent/search_agent.py+5-6
- python/tvm/contrib/msc/core/gym/control/configer.py+5-7
- python/tvm/contrib/msc/core/gym/control/controller.py+2-4
- python/tvm/contrib/msc/core/gym/control/service.py+28-18
- python/tvm/contrib/msc/core/gym/control/worker.py+18-14
- python/tvm/contrib/msc/core/gym/environment/base_env.py+83-22
- python/tvm/contrib/msc/core/gym/environment/method.py+9-5
- python/tvm/contrib/msc/core/gym/environment/prune_env.py+25-26
- python/tvm/contrib/msc/core/gym/environment/quantize_env.py+18-40
- python/tvm/contrib/msc/core/gym/namespace.py+40
- python/tvm/contrib/msc/core/runtime/hook.py+1-3
- python/tvm/contrib/msc/core/runtime/runner.py+39-4
- python/tvm/contrib/msc/core/tools/configer.py+10-10
- python/tvm/contrib/msc/core/tools/distill/distiller.py+1-3
- python/tvm/contrib/msc/core/tools/distill/method.py+4-2
- python/tvm/contrib/msc/core/tools/execute.py+1-1
- python/tvm/contrib/msc/core/tools/prune/method.py+4-2
- python/tvm/contrib/msc/core/tools/prune/pruner.py+1-3
- python/tvm/contrib/msc/core/tools/quantize/method.py+4-2
- python/tvm/contrib/msc/core/tools/quantize/quantizer.py+1-3
- python/tvm/contrib/msc/core/tools/track/configer.py+12-3
- python/tvm/contrib/msc/core/tools/track/method.py+1-3
- python/tvm/contrib/msc/core/tools/track/tracker.py+1-3
- python/tvm/contrib/msc/core/utils/expr.py+24-1
- python/tvm/contrib/msc/core/utils/file.py+24-2
- 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+76-105
- python/tvm/contrib/msc/framework/tensorflow/tools/distill/distiller.py+3-2
- python/tvm/contrib/msc/framework/tensorflow/tools/prune/pruner.py+3-2
- python/tvm/contrib/msc/framework/tensorflow/tools/quantize/quantizer.py+3-2
- python/tvm/contrib/msc/framework/tensorflow/tools/track/tracker.py+3-2
- python/tvm/contrib/msc/framework/tensorrt/runtime/runner.py+23
- python/tvm/contrib/msc/framework/tensorrt/tools/distill/distiller.py+3-2
- python/tvm/contrib/msc/framework/tensorrt/tools/prune/pruner.py+3-2
- python/tvm/contrib/msc/framework/tensorrt/tools/quantize/method.py+1-3
- python/tvm/contrib/msc/framework/tensorrt/tools/quantize/quantizer.py+3-2
- python/tvm/contrib/msc/framework/tensorrt/tools/track/tracker.py+3-2
- python/tvm/contrib/msc/framework/torch/tools/distill/distiller.py+3-2
- python/tvm/contrib/msc/framework/torch/tools/distill/method.py+1-3
- python/tvm/contrib/msc/framework/torch/tools/prune/pruner.py+3-2
- python/tvm/contrib/msc/framework/torch/tools/quantize/method.py+1-3
- python/tvm/contrib/msc/framework/torch/tools/quantize/quantizer.py+3-2
- python/tvm/contrib/msc/framework/torch/tools/track/tracker.py+3-2
- python/tvm/contrib/msc/framework/tvm/runtime/runner.py+34-4
- python/tvm/contrib/msc/framework/tvm/tools/distill/distiller.py+3-2
- python/tvm/contrib/msc/framework/tvm/tools/prune/pruner.py+3-2
- python/tvm/contrib/msc/framework/tvm/tools/quantize/method.py+1-3
- python/tvm/contrib/msc/framework/tvm/tools/quantize/quantizer.py+3-2
- python/tvm/contrib/msc/framework/tvm/tools/track/tracker.py+3-2
- python/tvm/contrib/msc/pipeline/config.py+8-6
- python/tvm/contrib/msc/pipeline/manager.py+101-39
- python/tvm/contrib/msc/pipeline/wrapper.py+16-10
- python/tvm/dlight/gpu/low_batch_gemv.py+1-1
- python/tvm/relax/frontend/nn/op.py+97-3
- python/tvm/relax/frontend/onnx/onnx_frontend.py+23-1
- python/tvm/relax/op/nn/__init__.py+2
- python/tvm/relax/op/nn/nn.py+109
- python/tvm/relax/transform/lazy_transform_params.py+1-1
- python/tvm/relax/transform/legalize_ops/nn.py+56
- python/tvm/runtime/disco/session.py+41
- python/tvm/script/parser/relax/entry.py+24
- python/tvm/script/parser/relax/parser.py+24-11
- python/tvm/topi/nn/pooling.py+7
- src/relax/op/nn/pooling.cc+173-8
- src/script/printer/relax/binding.cc+18
- src/support/libinfo.cc+1
- src/topi/nn.cc+5
- tests/python/dlight/test_gpu_matmul_tensorize.py+4
- tests/python/relax/test_frontend_nn_op.py+9-3
- tests/python/relax/test_frontend_onnx.py+6-1
- tests/python/relax/test_transform_lazy_transform_params.py+20
- tests/python/relax/test_tvmscript_parser.py+71
- tests/python/relax/test_tvmscript_printer_relax.py+10-4
- tests/python/tvmscript/test_tvmscript_roundtrip.py+43-3
0 commit comments