OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[Profiler][HPU] Enable profiler.key_averages().table() for HPU devices (#150770)

Browse Source 
Fixes #150769

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150770
Approved by: https://github.com/sraikund16, https://github.com/jeromean
This commit is contained in:
wdziurdz 2025-04-11 05:17:08 +00:00 committed by PyTorch MergeBot
parent 85ada5d6dd
commit 6c7336cb31
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/autograd/profiler_util.py
View File

@ -609,6 +609,7 @@ class FunctionEvent(FormattedTimesMixin):
DeviceType.CUDA,
DeviceType.PrivateUse1,
DeviceType.MTIA,
DeviceType.HPU,
]
return self.time_range.elapsed_us()
@ -633,6 +634,7 @@ class FunctionEvent(FormattedTimesMixin):
DeviceType.CUDA,
DeviceType.PrivateUse1,
DeviceType.MTIA,
DeviceType.HPU,
]
return self.device_time_total