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 2021/12/15 04:24:36 UTC

[tvm] branch main updated (21abb6e -> dd42ef2)

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 21abb6e  Fix for tvm.build()'s name warning (#9678)
     add dd42ef2  [CUTLASS] Add conv2d profiler (#9737)

No new revisions were added by this update.

Summary of changes:
 python/tvm/contrib/cutlass/build.py           |  12 +-
 python/tvm/contrib/cutlass/conv2d_profiler.py | 163 ++++++++++++++++++++++++++
 python/tvm/contrib/cutlass/gen_conv2d.py      |  78 +++++++++---
 python/tvm/contrib/cutlass/gen_gemm.py        |  24 ++--
 python/tvm/contrib/cutlass/gen_tensor_op.py   |  16 +--
 python/tvm/relay/op/strategy/cuda.py          |   4 +-
 tests/python/contrib/test_cutlass.py          |  18 +--
 7 files changed, 265 insertions(+), 50 deletions(-)
 create mode 100644 python/tvm/contrib/cutlass/conv2d_profiler.py