Skip to content

Commit

Permalink
[XPU] host timer check version from Torch 2.5 to Torch 2.6 (#6633)
Browse files Browse the repository at this point in the history
Elapsed time would be supported in Torch 2.6.

Co-authored-by: Masahiro Tanaka <[email protected]>
  • Loading branch information
YizhouZ and tohtana authored Oct 22, 2024
1 parent 40bde52 commit 11bbf45
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions accelerator/xpu_accelerator.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@ def is_synchronized_device(self):
return False

def use_host_timers(self):
# WA XPU event will be consolidated in 2.5
if ipex.__version__ < '2.5':
# WA XPU event will be consolidated in 2.6
if ipex.__version__ < '2.6':
return True
else:
return self.is_synchronized_device()
Expand Down

0 comments on commit 11bbf45

Please sign in to comment.