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/10/29 02:09:24 UTC
[tvm] branch main updated (dae71db -> 541f9f2)
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 dae71db [microTVM] Fix AOT/ARMv7m tests on physical devices. (#9364)
add 541f9f2 [BYOC] CUTLASS integration (#9261)
No new revisions were added by this update.
Summary of changes:
.gitmodules | 3 +
3rdparty/cutlass | 1 +
CMakeLists.txt | 2 +
LICENSE | 5 +
.../modules/contrib/CUTLASS.cmake | 11 +-
licenses/LICENSE.cutlass.txt | 23 ++
.../{_ffi/_ctypes => contrib/cutlass}/__init__.py | 3 +-
python/tvm/contrib/cutlass/build.py | 172 ++++++++++
python/tvm/contrib/cutlass/gemm_operation.py | 262 +++++++++++++++
python/tvm/contrib/cutlass/gemm_profiler.py | 196 +++++++++++
python/tvm/contrib/cutlass/gen_gemm.py | 355 ++++++++++++++++++++
python/tvm/contrib/cutlass/library.py | 219 ++++++++++++
python/tvm/relay/op/contrib/__init__.py | 1 +
python/tvm/relay/op/contrib/cutlass.py | 74 +++++
src/relay/backend/contrib/codegen_c/codegen_c.h | 6 +
src/relay/backend/contrib/cutlass/codegen.cc | 369 +++++++++++++++++++++
src/relay/backend/contrib/dnnl/codegen.cc | 6 -
tests/python/contrib/test_cutlass.py | 135 ++++++++
18 files changed, 1830 insertions(+), 13 deletions(-)
create mode 160000 3rdparty/cutlass
copy conda/recipe/conda_build_config.yaml => cmake/modules/contrib/CUTLASS.cmake (79%)
create mode 100644 licenses/LICENSE.cutlass.txt
copy python/tvm/{_ffi/_ctypes => contrib/cutlass}/__init__.py (89%)
create mode 100644 python/tvm/contrib/cutlass/build.py
create mode 100644 python/tvm/contrib/cutlass/gemm_operation.py
create mode 100644 python/tvm/contrib/cutlass/gemm_profiler.py
create mode 100644 python/tvm/contrib/cutlass/gen_gemm.py
create mode 100644 python/tvm/contrib/cutlass/library.py
create mode 100644 python/tvm/relay/op/contrib/cutlass.py
create mode 100644 src/relay/backend/contrib/cutlass/codegen.cc
create mode 100644 tests/python/contrib/test_cutlass.py