You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2023/02/17 01:52:07 UTC
[tvm] branch main updated (0c965f442a -> d7253fbc7e)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 0c965f442a [ONNX] Support ScatterElements with reduction (#13894)
add d7253fbc7e [TIR] Add cp.async support for tir.if_then_else (#13966)
No new revisions were added by this update.
Summary of changes:
src/target/source/codegen_cuda.cc | 8 +-
src/target/source/ptx.cc | 31 +++
src/target/source/ptx.h | 16 ++
src/tir/transforms/inject_ptx_async_copy.cc | 156 ++++++++------
.../unittest/test_cp_async_in_if_then_else.py | 238 +++++++++++++++++++++
5 files changed, 386 insertions(+), 63 deletions(-)
create mode 100644 tests/python/unittest/test_cp_async_in_if_then_else.py