You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by co...@apache.org on 2021/12/16 18:49:31 UTC
[tvm] branch main updated (c7ddb41 -> aa86dc0)
This is an automated email from the ASF dual-hosted git repository.
comaniac pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from c7ddb41 [Relay] Add a unit test for structural equality (#9745)
add aa86dc0 [CUTLASS] Support conv2d activation fusion (#9746)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cutlass/build.py | 9 ++++
python/tvm/contrib/cutlass/conv2d_operation.py | 35 ++++++++++----
python/tvm/contrib/cutlass/gen_conv2d.py | 9 ++--
python/tvm/contrib/cutlass/gen_gemm.py | 1 -
python/tvm/contrib/cutlass/library.py | 2 +
python/tvm/relay/op/contrib/cutlass.py | 55 ++++++++++++++++++---
src/relay/backend/contrib/cutlass/codegen.cc | 56 ++++++++++++++++++++--
tests/python/contrib/test_cutlass.py | 66 +++++++++++++++++++++-----
8 files changed, 196 insertions(+), 37 deletions(-)