applying the changes from net_ib.cc to rocm_net_ib.cc to ensure DMABUF-disabled configurations are respected. (#2152)

Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
此提交包含在:
mberenjk
2026-01-21 12:11:56 -08:00
提交者 GitHub
父節點 9a443f3054
當前提交 3d4813d991
+1 -1
查看文件
@@ -2116,7 +2116,7 @@ ib_recv:
}
}
useDmaBuf = (rocmIbDmaBufSupport(lComm->dev) == ncclSuccess);
useDmaBuf = (rocmIbDmaBufSupport(lComm->dev) == ncclSuccess && ncclParamDmaBufEnable());
rComm->flushEnabled = ((rocmIbGdrSupport() == ncclSuccess || useDmaBuf)
&& (ncclIbGdrFlushDisable == 0)) ? 1 : 0;
for (int i = 0; i < rComm->base.vProps.ndevs; i++) {