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 74715f4 [AutoScheduler] Fix task extraction with TE compiler (#8560)
add b6b8269 add support for softmax and log_softmax with MIOpen (#8543)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/miopen.py | 52 +++++++++++++++++
python/tvm/relay/op/strategy/rocm.py | 39 +++++++++++++
src/runtime/contrib/miopen/miopen_utils.cc | 4 ++
src/runtime/contrib/miopen/miopen_utils.h | 7 +++
src/runtime/contrib/miopen/softmax.cc | 92 ++++++++++++++++++++++++++++++
tests/python/contrib/test_miopen.py | 66 ++++++++++++++++++++-
6 files changed, 257 insertions(+), 3 deletions(-)
create mode 100644 src/runtime/contrib/miopen/softmax.cc
