This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 2dad4d1 [BugFix] Fix divide by zero error in TIR pass
lower_warp_memory (#9485)
add 3ad7c4a [Conv2DTransposed] Fix wrong shape check and add new TOPI
module to support groups (#9465)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/strategy/cuda.py | 2 +-
python/tvm/relay/op/strategy/generic.py | 26 ++--
python/tvm/relay/op/strategy/x86.py | 18 ++-
python/tvm/topi/generic/nn.py | 17 +++
python/tvm/topi/nn/conv2d_transpose.py | 129 +++++++++++++++++
python/tvm/topi/testing/conv2d_transpose_python.py | 40 +++++-
src/relay/op/nn/convolution.h | 21 +--
.../python/test_topi_group_conv2d_transpose.py | 156 +++++++++++++++++++++
8 files changed, 385 insertions(+), 24 deletions(-)
create mode 100644 tests/python/topi/python/test_topi_group_conv2d_transpose.py
