You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2022/04/08 22:05:49 UTC
[tvm] branch last-successful updated (0c17f07aa7 -> 81d72e3d63)
This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 0c17f07aa7 [ci] Remove hardcoded test shards (#10743)
add 81d72e3d63 [CUDNN] Add cuDNN as a Relay partitioning target (BYOC) (#10871)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/contrib/cublas.py | 54 +++----------------
python/tvm/relay/op/contrib/cudnn.py | 89 ++++++++++++++++++++++++++++++++
python/tvm/relay/op/contrib/te_target.py | 70 +++++++++++++++++++++++++
tests/python/contrib/test_cudnn.py | 68 ++++++++++++++++++++++++
tests/scripts/task_mypy.sh | 4 +-
5 files changed, 237 insertions(+), 48 deletions(-)
create mode 100644 python/tvm/relay/op/contrib/cudnn.py
create mode 100644 python/tvm/relay/op/contrib/te_target.py