Disable ipex env var if false (#24885)

Disable ipex if in use

Disable ipex env var if false (#24885)
Disable ipex if in use
a982c022 · Zach Mueller · GitHub · 07360b6c · a982c022
Unverified Commit a982c022 authored Jul 18, 2023 by Zach Mueller Committed by GitHub Jul 18, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/transformers/training_args.py src/transformers/training_args.py +2 -0

No files found.
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -1718,6 +1718,8 @@ class TrainingArguments:
                )
            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