Skip to content

Commit d933327

Browse files
jataylopruthvistony
authored andcommitted
Update clock info metric AMDSMI (#1459)
1 parent c33b3c8 commit d933327

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

torch/cuda/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1120,7 +1120,11 @@ def _get_amdsmi_power_draw(device: Optional[Union[Device, int]] = None) -> int:
11201120

11211121
def _get_amdsmi_clock_rate(device: Optional[Union[Device, int]] = None) -> int:
11221122
handle = _get_amdsmi_handler(device)
1123-
return amdsmi.amdsmi_get_clock_info(handle, amdsmi.AmdSmiClkType.GFX)["cur_clk"]
1123+
clock_info = amdsmi.amdsmi_get_clock_info(handle, amdsmi.AmdSmiClkType.GFX)
1124+
if "cur_clk" in clock_info:
1125+
return clock_info["cur_clk"]
1126+
else:
1127+
return clock_info["clk"]
11241128

11251129

11261130
def memory_usage(device: Optional[Union[Device, int]] = None) -> int:

0 commit comments

Comments
 (0)