Check fine grained memory before enabling RDMA

Adding back the check which was lost from 2.5 merge.


[ROCm/rccl commit: fb59328a7b]
此提交包含在:
Wenkai Du committed 2020-03-02 11:18:27 -08:00
1 父節點 2a66deb694
當前提交 593d99d9a9
1 file changed
+8
+8
查看文件
@@ -61,6 +61,14 @@ NCCL_PARAM(NetGdrLevel, "NET_GDR_LEVEL", PATH_PHB);
static ncclResult_t netGetGdrSupport(struct ncclTopoSystem* topo, int64_t busId, int netDev, int read, int* useGdr) {
*useGdr = 0;
int cudaDev;
CUDACHECK(hipGetDevice(&cudaDev));
if (!hasFineGrainVramPcie()) {
INFO(NCCL_INIT|NCCL_NET,"NET/%s : GPU Direct RDMA Disabled for GPU %d / Need Fine Grain VRAM over PCIe", ncclNetName(), cudaDev);
return ncclSuccess;
}
if (read) { // For reads (sends) only enable under certain conditions
int gdrReadParam = ncclParamNetGdrRead();
if (gdrReadParam == 0) return ncclSuccess;