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 à :
@@ -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++) {
|
||||||
|
|||||||
Référencer dans un nouveau ticket
Bloquer un utilisateur