From a982c0225efe3af4700f09d09890b239f551870c Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Tue, 18 Jul 2023 16:07:02 -0400 Subject: [PATCH] Disable ipex env var if false (#24885) Disable ipex if in use --- src/transformers/training_args.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index 3a30adbe0b..17adfc3355 100644 --- 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