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 2023/01/05 05:34:22 UTC

[tvm] branch main updated (bf0607bd31 -> 07a5a9eadb)

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 bf0607bd31 [VTA] Provide zero-initialization for VTAGenericInsn (#13698)
     add 07a5a9eadb [Tensorize][runtime] Add support for AMX(Advanced Matrix Extensions) through Tensor intrinsics (#13642)

No new revisions were added by this update.

Summary of changes:
 CMakeLists.txt                                     |   2 +
 cmake/config.cmake                                 |   3 +
 cmake/modules/LibInfo.cmake                        |   1 +
 cmake/modules/contrib/{CUTLASS.cmake => AMX.cmake} |  10 +-
 python/tvm/relay/op/strategy/x86.py                |   9 +-
 python/tvm/topi/x86/dense.py                       | 167 ++++++++++++---
 python/tvm/topi/x86/dense_alter_op.py              |  10 +-
 python/tvm/topi/x86/tensor_intrin.py               | 226 ++++++++++++++++++++-
 python/tvm/topi/x86/utils.py                       |   7 +
 src/runtime/contrib/amx/amx_config.cc              | 135 ++++++++++++
 src/runtime/thread_storage_scope.h                 |   5 +
 src/support/libinfo.cc                             |   5 +
 tests/python/contrib/test_amx.py                   | 126 ++++++++++++
 tests/python/relay/test_op_level1.py               |  47 +++++
 14 files changed, 714 insertions(+), 39 deletions(-)
 copy cmake/modules/contrib/{CUTLASS.cmake => AMX.cmake} (75%)
 create mode 100644 src/runtime/contrib/amx/amx_config.cc
 create mode 100644 tests/python/contrib/test_amx.py