This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 02eb183 [Relay][Topi][AutoTVM] Winograd support for Conv3D (#5186)
add 799ff35 [Runtime][Contrib] Support cudnn softmax (#5214)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cudnn.py | 24 ++++++++
python/tvm/relay/op/nn/_nn.py | 4 +-
python/tvm/relay/op/op_attrs.py | 5 ++
python/tvm/relay/op/strategy/cuda.py | 22 +++++++-
python/tvm/relay/op/strategy/generic.py | 22 +++++++-
python/tvm/relay/op/strategy/hls.py | 16 +++++-
python/tvm/relay/op/strategy/opengl.py | 16 +++++-
python/tvm/relay/op/strategy/x86.py | 16 +++++-
src/relay/op/nn/nn.cc | 9 +--
src/runtime/contrib/cudnn/cudnn_utils.cc | 10 ++++
src/runtime/contrib/cudnn/cudnn_utils.h | 8 +++
src/runtime/contrib/cudnn/softmax.cc | 94 ++++++++++++++++++++++++++++++++
tests/python/contrib/test_cudnn.py | 46 ++++++++++++++++
topi/python/topi/cuda/__init__.py | 2 +-
topi/python/topi/cuda/softmax.py | 12 ++++
topi/python/topi/nn/softmax.py | 1 -
16 files changed, 281 insertions(+), 26 deletions(-)
create mode 100644 src/runtime/contrib/cudnn/softmax.cc
