Use NUMA distance for hop count calculation

[ROCm/clr commit: 56d2dc0022]
2019-05-15 21:50:35 +00:00
parent 6ce0a62bb5
commit 3d75b10e0b
1 changed files with 4 additions and 3 deletions
@@ -574,9 +574,10 @@ hipError_t hipExtGetLinkTypeAndHopCount(int device1, int device2, uint32_t* link
        HSA_ERROR_CHECK(err);
        *linktype = link_info.link_type;

-        err = hsa_amd_agent_memory_pool_get_info(device1Handle->_hsaAgent, pool, HSA_AMD_AGENT_MEMORY_POOL_INFO_NUM_LINK_HOPS, hopcount);
-        HSA_ERROR_CHECK(err);
-
+        if (link_info.numa_distance < 30)
+            *hopcount = 1;
+        else
+            *hopcount = 2;
        return ihipLogStatus(hipSuccess);
    }
 }