You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ar...@apache.org on 2021/05/20 23:59:27 UTC
[tvm] branch main updated (71ff875 -> 28ea03c)
This is an automated email from the ASF dual-hosted git repository.
areusch pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 71ff875 [AOT] Remove lookup parameter function in AOT (#7988)
add 28ea03c [TOPI] Custom schedule for standalone transpose in cuda (#8030)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/_transform.py | 4 +-
python/tvm/relay/op/strategy/cuda.py | 22 ++++++++
python/tvm/relay/op/strategy/generic.py | 7 +++
python/tvm/topi/cuda/__init__.py | 1 +
python/tvm/topi/cuda/sparse.py | 39 +-------------
python/tvm/topi/cuda/transform.py | 67 +++++++++++++++++++++++++
tests/python/topi/python/test_topi_transform.py | 26 ++++++++++
vta/tutorials/autotvm/tune_relay_vta.py | 2 +-
8 files changed, 129 insertions(+), 39 deletions(-)
create mode 100644 python/tvm/topi/cuda/transform.py