AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

Use ncclSend/ncclRecv for alltoall type of collectives as default (#297)

Telusuri Sumber
This commit is contained in:
Wenkai Du
2020-11-09 11:23:17 -08:00
committed by GitHub
orang tua bdd8adf1ca
melakukan 2e8b3a0857
1 mengubah file dengan 1 tambahan dan 1 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
src/init.cc
+1 -1
Melihat File
@@ -299,7 +299,7 @@ static ncclResult_t commFree(ncclComm_t comm) {
return ncclSuccess;
}
RCCL_PARAM(AllToAllDisable, "ALLTOALL_KERNEL_DISABLE", 0);
RCCL_PARAM(AllToAllDisable, "ALLTOALL_KERNEL_DISABLE", 1);
static ncclResult_t commAlloc(ncclComm_t* comret, int ndev, int rank) {
if (ndev < 1) {