You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2022/02/08 01:29:06 UTC
[tvm] branch main updated (2ea2f5a -> 7fd73b2)
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 2ea2f5a [skip ci] Fix scipy intersphinx link (#10181)
add 7fd73b2 [CUTLASS] Initial support for conv2d wgrad (#10177)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cudnn.py | 14 +-
python/tvm/contrib/cutlass/gen_conv2d.py | 2 +
python/tvm/contrib/cutlass/gen_gemm.py | 4 +
python/tvm/contrib/cutlass/gen_tensor_op.py | 19 ++-
python/tvm/contrib/cutlass/library.py | 2 +
python/tvm/relay/op/contrib/cutlass.py | 13 ++
python/tvm/relay/op/nn/_nn.py | 35 ++++
python/tvm/topi/cuda/conv2d.py | 8 +-
.../topi/testing/conv2d_backcward_weight_python.py | 43 ++++-
src/relay/backend/contrib/cutlass/codegen.cc | 5 +
src/relay/op/nn/convolution.cc | 25 ++-
tests/python/contrib/test_cutlass.py | 180 ++++++++++++++++++---
tests/python/relay/test_op_grad_level2.py | 27 +++-
13 files changed, 335 insertions(+), 42 deletions(-)