[Intel GPU] Add synchronize() in torch.utils.benchmark

DDEle · DDEle · commit ddaace6c26f5 · 2025-02-25T08:39:05.000Z
diff --git a/torch/utils/benchmark/utils/timer.py b/torch/utils/benchmark/utils/timer.py
@@ -17,6 +17,10 @@
     def timer() -> float:
         torch.cuda.synchronize()
         return timeit.default_timer()
+elif torch.xpu.is_available():
+    def timer() -> float:
+        torch.xpu.synchronize()
+        return timeit.default_timer()
 elif torch._C._get_privateuse1_backend_name() != "privateuseone":
     privateuse1_device_handler = getattr(torch, torch._C._get_privateuse1_backend_name(), None) \
         if torch._C._get_privateuse1_backend_name() != "cpu" else None
