This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 44a1d1f Better reflect allocator names in CRT tests (#8828)
add 2859c20 [M3a][Meta Schedule] Add Sampling Primitive
SampleCategorical. (#8817)
No new revisions were added by this update.
Summary of changes:
include/tvm/support/random_engine.h | 10 ++-
include/tvm/tir/schedule/schedule.h | 27 +++++--
python/tvm/tir/schedule/schedule.py | 57 ++++++++++++++
src/support/array.h | 68 +++++++++++++++++
src/tir/schedule/concrete_schedule.cc | 30 +++++++-
src/tir/schedule/concrete_schedule.h | 44 +++++------
src/tir/schedule/primitive.h | 14 ++++
src/tir/schedule/primitive/sampling.cc | 88 +++++++++++++++++++++
src/tir/schedule/schedule.cc | 18 +++--
src/tir/schedule/traced_schedule.cc | 18 ++++-
src/tir/schedule/traced_schedule.h | 10 +++
.../python/unittest/test_tir_schedule_sampling.py | 89 ++++++++++++++++++++++
12 files changed, 429 insertions(+), 44 deletions(-)
create mode 100644 src/tir/schedule/primitive/sampling.cc
create mode 100644 tests/python/unittest/test_tir_schedule_sampling.py
