pin ipex

Author: IlyasMoutawwakil

optimum_benchmark/backends/ipex/backend.py

@@ -1,3 +1,4 @@
+import os
 from collections import OrderedDict
 from tempfile import TemporaryDirectory
 from typing import Any, Dict
@@ -17,6 +18,9 @@
 if is_torch_distributed_available():
     import torch.distributed
 
+if not hasattr(os, "exit"):
+    os.exit = os._exit
+
 
 class IPEXBackend(Backend[IPEXConfig]):
     NAME: str = "ipex"

pyproject.toml

@@ -85,7 +85,7 @@ optimum-benchmark = "optimum_benchmark.cli:main"
 
 [project.optional-dependencies]
 # optimum backends
-ipex = ["optimum-intel[ipex]>=1.25.0"]
+ipex = ["optimum-intel[ipex]>=1.25.0", "intel-extension-for-pytorch>=2.8.0", "torch>=2.8.0"]
 openvino = ["optimum-intel[openvino,nncf]>=1.25.0"]
 onnxruntime-gpu = ["optimum[onnxruntime-gpu]>=1.27.0"]
 onnxruntime = ["optimum[onnxruntime]>=1.27.0"]