Fix AOT compilation failed in Test with pip workflow (#3135)

Use g++ instead of icpx when DLE/PTDB/ONEAPI not installed.

Information about `sycl` usage in Triton XPU.

![image](https://github.com/user-attachments/assets/4ea4b253-9785-454c-b8df-072b7b7a86bb)

---------

Co-authored-by: Anatoly Myachev <[email protected]>

Author: yudongsi

python/test/unit/tools/test_aot.py

@@ -102,19 +102,16 @@ def kernel(C, A, B, M, N, K,
 def gen_kernel_library_xpu(dir, libname):
     cpp_files = glob.glob(os.path.join(dir, "*.cpp"))
     subprocess.run(
-        ["icpx"] + cpp_files + ["-I", COMPILATION_HELPER.include_dir[0], "-c", "-fsycl", "-fPIC"],
+        ["g++"] + cpp_files + ["-I" + include_dir for include_dir in COMPILATION_HELPER.include_dir] +
+        ["-L" + COMPILATION_HELPER.libsycl_dir, "-c", "-lsycl", "-fPIC"],
         check=True,
         cwd=dir,
     )
     o_files = glob.glob(os.path.join(dir, "*.o"))
 
-    command = ["icpx", "-fsycl", "-lze_loader", *o_files, "-shared", "-o", libname]
-    for lib_dir in COMPILATION_HELPER.library_dir:
-        command.extend(["-L", lib_dir])
-    if COMPILATION_HELPER.libsycl_dir:
-        for lib_dir in COMPILATION_HELPER.libsycl_dir:
-            command.extend(["-L", lib_dir])
-    subprocess.run(command, check=True, cwd=dir)
+    subprocess.run(["g++"] + [*o_files, "-shared", "-o", libname] +
+                   ["-L" + library_dir for library_dir in COMPILATION_HELPER.library_dir] +
+                   ["-L" + COMPILATION_HELPER.libsycl_dir, "-lsycl", "-lze_loader"], check=True, cwd=dir)
 
 
 def gen_kernel_library(dir, libname):
@@ -297,15 +294,15 @@ def gen_test_bin(dir, M, N, K, exe="test", algo_id=0):
         command.extend(["-l", "cuda", "-L", dir, "-l", "kernel", "-o", exe])
 
     if is_xpu():
-        command = ["icpx", "test.cpp"]
+        command = ["g++", "test.cpp"]
         for inc_dir in COMPILATION_HELPER.include_dir:
             command.extend(["-I", inc_dir])
         for lib_dir in COMPILATION_HELPER.library_dir:
             command.extend(["-L", lib_dir])
         if COMPILATION_HELPER.libsycl_dir:
             for lib_dir in COMPILATION_HELPER.libsycl_dir:
                 command.extend(["-L", lib_dir])
-        command.extend(["-fsycl", "-lze_loader", "-L", dir, "-l", "kernel", "-o", exe])
+        command.extend(["-lsycl", "-lze_loader", "-L", dir, "-l", "kernel", "-o", exe])
     subprocess.run(command, check=True, cwd=dir)
 
 

third_party/intel/backend/driver.py

@@ -6,15 +6,14 @@
 import tempfile
 from pathlib import Path
 from functools import cached_property
-from typing import Optional
 
 from triton.runtime.build import _build
 from triton.runtime.cache import get_cache_manager
 from triton.backends.compiler import GPUTarget
 from triton.backends.driver import DriverBase
 
 
-def find_sycl(include_dir: list[str]) -> tuple[list[str], Optional[str]]:
+def find_sycl(include_dir: list[str]) -> tuple[list[str], str]:
     """
     Looks for the sycl library in known places.
 
@@ -28,21 +27,26 @@ def find_sycl(include_dir: list[str]) -> tuple[list[str], Optional[str]]:
       AssertionError: if library was not found.
     """
     include_dir = include_dir.copy()
+    sycl_dir = None
     assertion_message = ("sycl headers not found, please install `icpx` compiler, "
                          "or provide `ONEAPI_ROOT` environment "
                          "or install `intel-sycl-rt>=2025.0.0` wheel")
-
-    if shutil.which("icpx") and os.name != "nt":
+    icpx_path = shutil.which("icpx")
+    if icpx_path and os.name != "nt":
         # only `icpx` compiler knows where sycl runtime binaries and header files are
-        return include_dir, None
+        compiler_root = os.path.abspath(f"{icpx_path}/../..")
+        include_dir += [os.path.join(compiler_root, "include"), os.path.join(compiler_root, "include/sycl")]
+        sycl_dir = os.path.join(compiler_root, "lib")
+        return include_dir, sycl_dir
 
     oneapi_root = os.getenv("ONEAPI_ROOT")
     if oneapi_root:
         include_dir += [
             os.path.join(oneapi_root, "compiler/latest/include"),
             os.path.join(oneapi_root, "compiler/latest/include/sycl")
         ]
-        return include_dir, None
+        sycl_dir = os.path.join(oneapi_root, "compiler/latest/lib"),
+        return include_dir, sycl_dir
 
     try:
         sycl_rt = importlib.metadata.metadata("intel-sycl-rt")
@@ -52,7 +56,6 @@ def find_sycl(include_dir: list[str]) -> tuple[list[str], Optional[str]]:
     if sycl_rt.get("version", "0.0.0").startswith("2024"):
         raise AssertionError(assertion_message)
 
-    sycl_dir = None
     for f in importlib.metadata.files("intel-sycl-rt"):
         # sycl/sycl.hpp and sycl/CL/sycl.hpp results in both folders
         # being add: include and include/sycl.
@@ -122,7 +125,7 @@ def include_dir(self) -> list[str]:
         return self._include_dir
 
     @cached_property
-    def libsycl_dir(self) -> Optional[str]:
+    def libsycl_dir(self) -> str:
         self._compute_compilation_options_lazy
         return self._libsycl_dir