diff --git a/src/transport/net_ib_rocm.cc b/src/transport/net_ib_rocm.cc index 8f0b9b19a4..8cbe077c18 100644 --- a/src/transport/net_ib_rocm.cc +++ b/src/transport/net_ib_rocm.cc @@ -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++) {