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

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

[ROCm/rccl commit: 9a443f3054]
Este commit está contenido en:
mberenjk
2026-01-21 11:08:12 -08:00
cometido por GitHub
padre 244047310e
commit 7069fc936f
Se han modificado 2 ficheros con 2 adiciones y 1 borrados
+1
Ver fichero
@@ -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
Ver fichero
@@ -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++) {