hpu_accelerator: use torch.use_deterministic_algorithms (#6897)

formal API instead of hpu.setDeterministic
microsoft · Dec 20, 2024 · eea5304 · eea5304
1 parent 00ea0c4
commit eea5304
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/accelerator/hpu_accelerator.py b/accelerator/hpu_accelerator.py
@@ -21,8 +21,8 @@ def __init__(self):
         self.apply_hpu_workarounds()
         try:
             import habana_frameworks.torch.hpu as hpu
-            hpu.setDeterministic(True)
             self.hpu = hpu
+            torch.use_deterministic_algorithms(True)
         except ImportError as e:
             raise ValueError(
                 f"HPU_Accelerator requires habana_frameworks.torch.hpu, which is not installed on this system.")