fix

Signed-off-by: Anatoly Myachev <[email protected]>

benchmarks/setup.py

@@ -9,6 +9,11 @@
 
 import torch
 
+ipex_cmake_prefix_path = ""
+if os.getenv("USE_IPEX", "1") == "1":
+    import intel_extension_for_pytorch
+    ipex_cmake_prefix_path = f";{intel_extension_for_pytorch.cmake_prefix_path}"
+
 
 class CMakeBuild():
 
@@ -42,7 +47,7 @@ def build_extension(self):
             "Ninja",  # Ninja is much faster than make
             "-DCMAKE_MAKE_PROGRAM=" +
             ninja_dir,  # Pass explicit path to ninja otherwise cmake may cache a temporary path
-            f"-DCMAKE_PREFIX_PATH={torch.utils.cmake_prefix_path}",
+            f"-DCMAKE_PREFIX_PATH={torch.utils.cmake_prefix_path}{ipex_cmake_prefix_path}",
             "-DCMAKE_EXPORT_COMPILE_COMMANDS=ON",
             "-DCMAKE_ARCHIVE_OUTPUT_DIRECTORY=" + self.extdir,
             "-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + self.extdir,