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

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

[ROCm/rccl commit: 9a443f3054]
Dieser Commit ist enthalten in:
mberenjk
2026-01-21 11:08:12 -08:00
committet von GitHub
Ursprung 244047310e
Commit 7069fc936f
2 geänderte Dateien mit 2 neuen und 1 gelöschten Zeilen
+1
Datei anzeigen
@@ -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
Datei anzeigen
@@ -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++) {