Add initial support for intel Gaudi accelerators (#2121)

2024-11-22 20:22:23 -08:00
parent 66d4859acf
commit 865233e256
4 changed files with 10 additions and 7 deletions
@@ -278,10 +278,7 @@ def correctness_test(


 def synchronize(device):
-    if device == "cuda":
-        torch.cuda.synchronize()
-    elif device == "xpu":
-        torch.xpu.synchronize()
+    torch.get_device_module(device).synchronize()


 def latency_test_run_once(