This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from b1951a7811 [USMP] add missing const specifier for
global_const_workspace (#16999)
add 7463b37b88 [Metal] Support metal device profiling (#17025)
add 604fbbdf0e Support multinomial_from_uniform dispatch (#17010)
add cf2753eafd [Relax][UnitTest] Validate IRModule with multiple targets
(#16960)
add 7f7762d53a [DLight] Perf improvement for low_batch_gemv on Metal
(#17026)
No new revisions were added by this update.
Summary of changes:
.../tvm/relax/attrs/sampling.h | 30 +-
python/tvm/dlight/gpu/low_batch_gemv.py | 13 +-
python/tvm/relax/backend/__init__.py | 3 +-
python/tvm/relax/backend/dispatch_sampling.py | 94 ++++++
python/tvm/relax/backend/dispatch_sort_scan.py | 46 +--
python/tvm/relax/backend/utils.py | 55 +++-
python/tvm/relax/backend_tir/__init__.py | 3 +-
python/tvm/relax/backend_tir/cumsum.py | 8 +-
python/tvm/relax/backend_tir/sampling.py | 339 +++++++++++++++++++++
python/tvm/relax/frontend/nn/op.py | 46 +--
python/tvm/relax/op/__init__.py | 7 +-
python/tvm/relax/op/sampling.py | 87 ++++++
python/tvm/relax/pipeline.py | 1 +
python/tvm/script/ir_builder/relax/ir.py | 83 ++---
python/tvm/script/parser/tir/parser.py | 25 +-
python/tvm/target/detect_target.py | 4 +
src/relax/op/tensor/index.cc | 2 +-
src/relax/op/tensor/sampling.cc | 143 +++++++++
src/relax/op/tensor/sampling.h | 57 ++++
src/runtime/metal/metal_device_api.mm | 37 +++
tests/python/dlight/test_gpu_low_batch_gemv.py | 138 ++++-----
.../python/relax/test_backend_dispatch_sampling.py | 201 ++++++++++++
tests/python/relax/test_frontend_nn_op.py | 40 +--
tests/python/relax/test_op_sampling.py | 69 +++++
tests/python/relax/test_vm_build.py | 59 ++++
.../python/tvmscript/test_tvmscript_parser_tir.py | 24 ++
26 files changed, 1362 insertions(+), 252 deletions(-)
copy src/relax/transform/gradient_simplifier.h =>
include/tvm/relax/attrs/sampling.h (59%)
create mode 100644 python/tvm/relax/backend/dispatch_sampling.py
create mode 100644 python/tvm/relax/backend_tir/sampling.py
create mode 100644 python/tvm/relax/op/sampling.py
create mode 100644 src/relax/op/tensor/sampling.cc
create mode 100644 src/relax/op/tensor/sampling.h
create mode 100644 tests/python/relax/test_backend_dispatch_sampling.py
create mode 100644 tests/python/relax/test_op_sampling.py
