Adding a check to respect DMABUF being disabled by the user (#2076)

Co-authored-by: Marzieh Berenjkoub <mberenjk@.amd.com>

[ROCm/rccl commit: 9a443f3054]
This commit is contained in:
mberenjk
2026-01-21 11:08:12 -08:00
committato da GitHub
parent 244047310e
commit 7069fc936f
2 ha cambiato i file con 2 aggiunte e 1 eliminazioni
+1
Vedi File
@@ -30,6 +30,7 @@ extern ncclNet_t ncclNetIb;
extern ncclNet_t ncclNetSocket;
extern ncclResult_t rcclNetP2pPolicy(void* handle, int isP2p);
extern int64_t ncclParamDmaBufEnable();
#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
extern ncclNet_t rocmNetIb;
+1 -1
Vedi File
@@ -1990,7 +1990,7 @@ ib_recv:
}
}
useDmaBuf = (ncclIbDmaBufSupport(lComm->dev) == ncclSuccess);
useDmaBuf = (ncclIbDmaBufSupport(lComm->dev) == ncclSuccess && ncclParamDmaBufEnable());
rComm->flushEnabled = ((ncclIbGdrSupport() == ncclSuccess || useDmaBuf)
&& (ncclParamIbGdrFlushDisable() == 0)) ? 1 : 0;
for (int i = 0; i < rComm->base.vProps.ndevs; i++) {