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

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

[ROCm/rccl commit: 9a443f3054]
此提交包含在:
mberenjk
2026-01-21 11:08:12 -08:00
提交者 GitHub
父節點 244047310e
當前提交 7069fc936f
共有 2 個檔案被更改,包括 2 行新增1 行删除
+1
查看文件
@@ -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
查看文件
@@ -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++) {