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 a66186b [FIX] Fix depthwise conv2d on non-cuda GPU platforms (#8379)
add 2e47947 [cuDNN] Add support for log_softmax (#8369)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cudnn.py | 26 ++++++++++
python/tvm/relay/op/nn/_nn.py | 2 +-
python/tvm/relay/op/strategy/cuda.py | 22 ++++++--
python/tvm/relay/op/strategy/generic.py | 15 +++---
python/tvm/relay/op/strategy/hls.py | 15 ++++--
python/tvm/relay/op/strategy/x86.py | 15 ++++--
python/tvm/topi/cuda/softmax.py | 10 ++++
python/tvm/topi/nn/softmax.py | 5 +-
src/runtime/contrib/cudnn/softmax.cc | 91 +++++++++++++++++----------------
tests/python/contrib/test_cudnn.py | 28 +++++++---
10 files changed, 156 insertions(+), 73 deletions(-)
