Use compiler OpenMP headers in SYCL CI

Author: allnes

scripts/ci/sycl_x86_setup.py

@@ -58,7 +58,6 @@ def setup_linux_runtime() -> None:
             "g++-13",
             "gcc-13",
             "git",
-            "libomp-dev",
             "ocl-icd-opencl-dev",
             "opencl-headers",
             "python3",
@@ -200,15 +199,13 @@ def setup_unix_compilers(sycl_root: Path) -> None:
         if openmp_library is None:
             main_error("Unable to locate libiomp5.so for the SYCL toolchain")
 
-        search_roots = [
-            Path(capture([str(cxx_path), "-print-resource-dir"])),
-            sycl_root,
-            Path("/opt/intel/oneapi"),
-        ]
-        search_roots.extend(Path("/usr/lib").glob("llvm-*"))
-        openmp_header = find_file(search_roots, "omp.h")
-        if openmp_header is None:
-            main_error("Unable to locate omp.h for the SYCL toolchain")
+        resource_dir = Path(capture([str(cxx_path), "-print-resource-dir"]))
+        openmp_header = resource_dir / "include" / "omp.h"
+        if not openmp_header.is_file():
+            main_error(
+                "Unable to locate compiler-provided omp.h under "
+                f"{resource_dir / 'include'}"
+            )
 
         write_env("ITLABAI_OPENMP_FLAGS", "-fopenmp=libiomp5")
         write_env("ITLABAI_OPENMP_LIBRARY", str(openmp_library))