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 920c380 [Caffe Frontend] adding Reduction op (#8015)
add ff2c434 [CUTLASS] Support more kernels: int8, tf32, and 3xtf32 (#9899)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/cutlass/build.py | 91 +++++++++++-
python/tvm/contrib/cutlass/gen_conv2d.py | 29 +++-
python/tvm/contrib/cutlass/gen_gemm.py | 66 +++++++--
python/tvm/contrib/cutlass/gen_tensor_op.py | 210 +++++++++++++++++++--------
python/tvm/contrib/cutlass/library.py | 17 +++
python/tvm/relay/op/contrib/cutlass.py | 7 +-
src/relay/backend/contrib/cutlass/codegen.cc | 6 +-
tests/python/contrib/test_cutlass.py | 135 ++++++++++++++---
8 files changed, 445 insertions(+), 116 deletions(-)
