This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from f8c55db [TFLite] Added ability to infer shapes for arguments (#7293)
add 62f251b [TOPI] Make cumsum IR reusable, add thrust scan (#7303)
No new revisions were added by this update.
Summary of changes:
python/tvm/topi/cuda/nms.py | 151 +------------
python/tvm/topi/cuda/scan.py | 406 +++++++++++++++++++++++++++++++++++
python/tvm/topi/cuda/scatter.py | 6 +-
python/tvm/topi/cuda/sort.py | 5 +-
python/tvm/topi/cuda/sparse.py | 5 +-
python/tvm/topi/utils.py | 5 +
src/runtime/contrib/thrust/thrust.cu | 76 +++++++
tests/python/contrib/test_sort.py | 35 +--
tests/python/contrib/test_thrust.py | 123 +++++++++++
tests/python/relay/test_any.py | 48 +++++
10 files changed, 666 insertions(+), 194 deletions(-)
create mode 100644 python/tvm/topi/cuda/scan.py
create mode 100644 tests/python/contrib/test_thrust.py
