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 3ce87cba21 Fix includes of custom allreduce kernel (#16814)
add 64db9f78a0 [Runtime] Introduce MSCCLPP with NCCL equivalent interface
(#16804)
No new revisions were added by this update.
Summary of changes:
3rdparty/mscclpp/include/common.h | 107 +++++++
3rdparty/mscclpp/include/msccl.cuh | 323 ++++++++++++++++++++
3rdparty/mscclpp/include/msccl.h | 494 +++++++++++++++++++++++++++++++
CMakeLists.txt | 5 +-
cmake/modules/contrib/MSCCLPP.cmake | 50 ++++
src/runtime/contrib/mscclpp/allreduce.cu | 184 ++++++++++++
6 files changed, 1161 insertions(+), 2 deletions(-)
create mode 100644 3rdparty/mscclpp/include/common.h
create mode 100644 3rdparty/mscclpp/include/msccl.cuh
create mode 100644 3rdparty/mscclpp/include/msccl.h
create mode 100644 cmake/modules/contrib/MSCCLPP.cmake
create mode 100644 src/runtime/contrib/mscclpp/allreduce.cu
