frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] Return correct AMDSMI socket_power metric (#130331)

Browse Source 
Extending on the change in https://github.com/pytorch/pytorch/pull/127729

Depending on gcnArch the API to return socket power will change based on underlying gpu_metrics. This PR will handle both cases.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130331
Approved by: https://github.com/jeffdaily, https://github.com/eqy, https://github.com/malfet
This commit is contained in:
Jack Taylor
2024-07-17 01:58:55 +00:00
committed by PyTorch MergeBot
parent 03c660468e
commit e9023d57b0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/cuda/__init__.py
View File

@ -1108,7 +1108,11 @@ def _get_amdsmi_temperature(device: Optional[Union[Device, int]] = None) -> int:
def _get_amdsmi_power_draw(device: Optional[Union[Device, int]] = None) -> int:
handle = _get_amdsmi_handler(device)
return amdsmi.amdsmi_get_power_info(handle)["current_socket_power"]
socket_power = amdsmi.amdsmi_get_power_info(handle)["average_socket_power"]
if socket_power != "N/A":
return socket_power
else:
return amdsmi.amdsmi_get_power_info(handle)["current_socket_power"]
def _get_amdsmi_clock_rate(device: Optional[Union[Device, int]] = None) -> int: