Relax Intel build-tree OpenMP assumptions

Author: allnes

scripts/ci/sycl_x86_setup.py

@@ -40,18 +40,7 @@ def linux_native_cpu_toolchain_ready(toolchain_root: Path) -> bool:
         find_file([toolchain_root], "remangled-l64-signed_char.libspirv.bc")
         is not None
     )
-    has_openmp_runtime = (
-        find_file([toolchain_root], "libiomp5.so") is not None
-        or find_file([toolchain_root], "libomp.so") is not None
-    )
-    has_openmp_header = find_file([toolchain_root], "omp.h") is not None
-    return (
-        has_compiler
-        and has_runtime
-        and has_native_cpu_libspirv
-        and has_openmp_runtime
-        and has_openmp_header
-    )
+    return has_compiler and has_runtime and has_native_cpu_libspirv
 
 
 def configure_platform(platform: str) -> None:
@@ -252,8 +241,10 @@ def setup_unix_compilers(sycl_root: Path) -> None:
     library_roots = [sycl_root]
     sycl_runtime_dirs: list[Path] = []
 
+    source_built_linux_toolchain = False
     if require_env("RUNNER_OS") == "Linux":
         compiler_root = Path(require_env("SYCL_ROOT"))
+        source_built_linux_toolchain = not os.environ.get("TOOLCHAIN_ASSET")
         cc_candidates = [
             compiler_root / "bin" / "icx",
             compiler_root / "bin" / "clang",
@@ -291,7 +282,7 @@ def setup_unix_compilers(sycl_root: Path) -> None:
             openmp_library = find_file([sycl_root, Path("/opt/intel/oneapi")], "libomp.so")
             openmp_library_name = "omp"
             openmp_flags = "-fopenmp"
-        if openmp_library is None:
+        if openmp_library is None and not source_built_linux_toolchain:
             main_error("Unable to locate an Intel toolchain OpenMP runtime library")
 
         resource_dir = Path(capture([str(cxx_path), "-print-resource-dir"]))
@@ -303,15 +294,16 @@ def setup_unix_compilers(sycl_root: Path) -> None:
             ],
             "omp.h",
         )
-        if openmp_header is None:
+        if openmp_header is None and not source_built_linux_toolchain:
             main_error(
                 "Unable to locate Intel-provided omp.h under the SYCL toolchain"
             )
 
-        write_env("ITLABAI_OPENMP_FLAGS", openmp_flags)
-        write_env("ITLABAI_OPENMP_LIBRARY_NAME", openmp_library_name)
-        write_env("ITLABAI_OPENMP_LIBRARY", str(openmp_library))
-        write_env("ITLABAI_OPENMP_INCLUDE_DIR", str(openmp_header.parent))
+        if openmp_library is not None and openmp_header is not None:
+            write_env("ITLABAI_OPENMP_FLAGS", openmp_flags)
+            write_env("ITLABAI_OPENMP_LIBRARY_NAME", openmp_library_name)
+            write_env("ITLABAI_OPENMP_LIBRARY", str(openmp_library))
+            write_env("ITLABAI_OPENMP_INCLUDE_DIR", str(openmp_header.parent))
 
         libsycl = find_file(library_roots + [sycl_root], "libsycl.so.8")
         if libsycl is None: