This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 3f064b6 Add conv1d support in BYOC TRT by converting conv1d to conv2d
(#9324)
add 9cf0245 [Relay, TOPI] Add searchsorted op (#9184)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/algorithm.h | 16 +++
python/tvm/relay/frontend/pytorch.py | 22 ++++
python/tvm/relay/op/_algorithm.py | 4 +
python/tvm/relay/op/algorithm.py | 34 ++++++
python/tvm/relay/op/op_attrs.py | 5 +
python/tvm/relay/op/strategy/cuda.py | 12 ++
python/tvm/relay/op/strategy/generic.py | 25 ++++
python/tvm/topi/__init__.py | 1 +
python/tvm/topi/cuda/__init__.py | 1 +
python/tvm/topi/cuda/searchsorted.py | 102 +++++++++++++++++
python/tvm/topi/searchsorted.py | 127 +++++++++++++++++++++
python/tvm/topi/testing/__init__.py | 1 +
.../__init__.py => topi/testing/searchsorted.py} | 36 +++---
src/relay/op/algorithm/searchsorted.cc | 86 ++++++++++++++
src/te/operation/create_primfunc.cc | 2 +-
tests/python/frontend/pytorch/test_forward.py | 30 +++++
tests/python/relay/test_op_level5.py | 1 -
tests/python/relay/test_op_level6.py | 24 ++++
tests/python/topi/python/test_topi_searchsorted.py | 93 +++++++++++++++
19 files changed, 602 insertions(+), 20 deletions(-)
create mode 100644 python/tvm/topi/cuda/searchsorted.py
create mode 100644 python/tvm/topi/searchsorted.py
copy python/tvm/{relay/frontend/__init__.py => topi/testing/searchsorted.py}
(52%)
create mode 100644 src/relay/op/algorithm/searchsorted.cc
create mode 100644 tests/python/topi/python/test_topi_searchsorted.py
