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 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
