Fix for weight/hops not being updated

Signed-off-by: Divya Shikre <DivyaUday.Shikre@amd.com>
Change-Id: I333d49fa011b85d41eca63c082c0615febe2f7e9


[ROCm/amdsmi commit: 94291bf882]
Этот коммит содержится в:
Divya Shikre
2020-10-20 14:59:04 -04:00
коммит произвёл Divya Uday Shikre
родитель 297f89a62a
Коммит 33ccef9a1e
+2 -2
Просмотреть файл
@@ -1914,7 +1914,6 @@ def showWeightTopology(deviceList):
"""
global PRINT_JSON
devices_ind = range(len(deviceList))
weight = c_uint64()
gpu_links_weight = [[0 for x in devices_ind] for y in devices_ind]
printLogSpacer(' Weight between two GPUs ')
for srcdevice in deviceList:
@@ -1922,6 +1921,7 @@ def showWeightTopology(deviceList):
if (srcdevice == destdevice):
gpu_links_weight[srcdevice][destdevice] = 0
continue
weight = c_uint64()
ret = rocmsmi.rsmi_topo_get_link_weight(srcdevice, destdevice, byref(weight))
if rsmi_ret_ok(ret):
gpu_links_weight[srcdevice][destdevice] = weight
@@ -1955,7 +1955,6 @@ def showHopsTopology(deviceList):
@param deviceList: List of DRM devices (can be a single-item list)
"""
hops = c_uint64()
linktype = c_char_p()
devices_ind = range(len(deviceList))
gpu_links_hops = [[0 for x in devices_ind] for y in devices_ind]
@@ -1965,6 +1964,7 @@ def showHopsTopology(deviceList):
if (srcdevice == destdevice):
gpu_links_hops[srcdevice][destdevice] = '0'
continue
hops = c_uint64()
ret = rocmsmi.rsmi_topo_get_link_type(srcdevice, destdevice, byref(hops), byref(linktype))
if rsmi_ret_ok(ret):
gpu_links_hops[srcdevice][destdevice] = hops