Skip to content

Commit dd43b9b

Browse files
authored
Update clock info metric AMDSMI (#1459)
1 parent 4459b67 commit dd43b9b

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
@@ -1106,7 +1106,11 @@ def _get_amdsmi_power_draw(device: Optional[Union[Device, int]] = None) -> int:
11061106

11071107
def _get_amdsmi_clock_rate(device: Optional[Union[Device, int]] = None) -> int:
11081108
handle = _get_amdsmi_handler(device)
1109-
return amdsmi.amdsmi_get_clock_info(handle, amdsmi.AmdSmiClkType.GFX)["cur_clk"]
1109+
clock_info = amdsmi.amdsmi_get_clock_info(handle, amdsmi.AmdSmiClkType.GFX)
1110+
if "cur_clk" in clock_info:
1111+
return clock_info["cur_clk"]
1112+
else:
1113+
return clock_info["clk"]
11101114

11111115

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

0 commit comments

Comments
 (0)