Disable ipex env var if false (#24885)

Disable ipex if in use
huggingface · Jul 18, 2023 · a982c02 · a982c02
1 parent 07360b6
commit a982c02
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py
@@ -1718,6 +1718,8 @@ def _setup_devices(self) -> "torch.device":
                 )
             AcceleratorState._reset_state(reset_partial_state=True)
         self.distributed_state = None
+        if not self.use_ipex and "ACCELERATE_USE_IPEX" not in os.environ:
+            os.environ["ACCELERATE_USE_IPEX"] = "false"
         if self.use_cpu:
             self.distributed_state = PartialState(cpu=True, backend=self.ddp_backend)
             self._n_gpu = 0