This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from a4cea71687 [Unity] Add batched A matrix support for FasterTransformer
fp16A intB matmul (#15501)
add 4ed34013de [Relax] CuDNN Fallback Support through BYOC. (#15462)
No new revisions were added by this update.
Summary of changes:
cmake/modules/CUDA.cmake | 2 +-
python/tvm/contrib/cudnn.py | 21 +-
python/tvm/relax/backend/contrib/cudnn.py | 105 +++++++++
python/tvm/relax/backend/patterns.py | 36 +++
python/tvm/relax/transform/transform.py | 2 +-
.../backend/contrib/{cublas => cudnn}/codegen.cc | 26 +--
src/runtime/contrib/cudnn/conv_backward.cc | 40 ++--
src/runtime/contrib/cudnn/conv_forward.cc | 36 +--
src/runtime/contrib/cudnn/cudnn_json_runtime.cc | 197 ++++++++++++++++
src/runtime/contrib/cudnn/cudnn_utils.h | 15 ++
src/runtime/contrib/json/json_node.h | 2 +-
tests/python/relax/test_codegen_cublas.py | 2 +-
tests/python/relax/test_codegen_cudnn.py | 252 +++++++++++++++++++++
13 files changed, 687 insertions(+), 49 deletions(-)
create mode 100644 python/tvm/relax/backend/contrib/cudnn.py
copy src/relax/backend/contrib/{cublas => cudnn}/codegen.cc (78%)
create mode 100644 src/runtime/contrib/cudnn/cudnn_json_runtime.cc
create mode 100644 tests/python/relax/test_codegen_cudnn.py
