AI/rocm-systems
2
0 Tükrözés 0
Kód Hibajegyek Egyesítési kérések Actions Packages Projektek Kiadások Wiki Tevékenység

Revert "msccl: build same number of kernels as in ROCm 5.7" (#1058)

Forráskód böngészése 
This reverts commit f960174d03be7e5174baa83b256526d388a38842.

[ROCm/rccl commit: 7987015a19]
This commit is contained in:
Wenkai Du authored and GitHub committed 2024-01-24 08:43:50 -08:00
1 szülő 36fdd214e0
commit 17f70df058
5 files changed
+9 -19
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/include/msccl/msccl_kernel.h
+1 -3
Fájl megtekintése
@@ -40,9 +40,7 @@ __global__ void MSCCL_KERNEL_ENTRY_NAME(devredop, type, proto, fullOps)(struct n
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(Sum, false) \
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(Prod, false) \
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(Max, false) \
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(Min, false) \
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(PreMulSum, false) \
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP_NOFLOAT(SumPostDiv, false)
MSCCL_DECL_KERNEL_ENTRY_FUNC_DEVREDOP(Min, false)
MSCCL_DECL_KERNEL_ENTRY_FUNC()