pytorch
pytorch copied to clipboard
[rocm6.3_internal_testing] [ROCm] [SWDEV-472265] Return correct AMDSMI socket_power metric
Copy of: https://github.com/ROCm/pytorch/pull/1457
Cherry pick of https://github.com/pytorch/pytorch/pull/130331/files
Extending on the change in https://github.com/pytorch/pytorch/pull/127729
Depending on gcnArch the API to return socket power will change. This will help us handle both cases for now until this is consolidated in the future. MI200 Current socket power: N/A Average socket power: 94
MI300 Current socket power: 717 Average socket power: N/A
Fixes https://ontrack-internal.amd.com/browse/SWDEV-472265