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>
This commit is contained in:
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++) {