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 72b22f8 [Relay][VM] Fix loading late bound consts when none exist
(#10087)
add a1f51aa [CUTLASS] Conv2d dgrad (#10110)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cutlass/build.py | 31 ++-
python/tvm/contrib/cutlass/conv2d_profiler.py | 18 +-
python/tvm/contrib/cutlass/gen_conv2d.py | 71 +++++-
python/tvm/contrib/cutlass/gen_tensor_op.py | 1 +
python/tvm/contrib/cutlass/library.py | 10 +
python/tvm/relay/op/contrib/cutlass.py | 32 ++-
python/tvm/relay/op/strategy/cuda.py | 31 ++-
python/tvm/relay/op/strategy/generic.py | 4 +-
python/tvm/topi/cuda/__init__.py | 2 +-
...onv2d_transpose_nchw.py => conv2d_transpose.py} | 14 +-
python/tvm/topi/nn/conv2d_transpose.py | 6 +
src/relay/backend/contrib/cutlass/codegen.cc | 93 +++++---
tests/python/contrib/test_cutlass.py | 238 ++++++++++++++++-----
tests/python/relay/test_op_level2.py | 38 +++-
14 files changed, 474 insertions(+), 115 deletions(-)
rename python/tvm/topi/cuda/{conv2d_transpose_nchw.py => conv2d_transpose.py}
(97%)
