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>
Cette révision appartient à :
mberenjk
2026-01-21 12:11:56 -08:00
révisé par GitHub
Parent 9a443f3054
révision 3d4813d991
+1 -1
Voir le fichier
@@ -2116,7 +2116,7 @@ ib_recv:
} }
} }
useDmaBuf = (rocmIbDmaBufSupport(lComm->dev) == ncclSuccess); useDmaBuf = (rocmIbDmaBufSupport(lComm->dev) == ncclSuccess && ncclParamDmaBufEnable());
rComm->flushEnabled = ((rocmIbGdrSupport() == ncclSuccess || useDmaBuf) rComm->flushEnabled = ((rocmIbGdrSupport() == ncclSuccess || useDmaBuf)
&& (ncclIbGdrFlushDisable == 0)) ? 1 : 0; && (ncclIbGdrFlushDisable == 0)) ? 1 : 0;
for (int i = 0; i < rComm->base.vProps.ndevs; i++) { for (int i = 0; i < rComm->base.vProps.ndevs; i++) {