Enable msvc to compile kernel launchers (#3185)

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

Author: anmyachev

python/triton/runtime/build.py

@@ -10,8 +10,8 @@ def is_xpu():
 
 
 def _cc_cmd(cc, src, out, include_dirs, library_dirs, libraries):
-    if cc in ["cl", "clang-cl"]:
-        cc_cmd = [cc, src, "/nologo", "/O2", "/LD"]
+    if "cl.EXE" in cc or "clang-cl" in cc:
+        cc_cmd = [cc, "/Zc:__cplusplus", src, "/nologo", "/O2", "/LD"]
         cc_cmd += [f"/I{dir}" for dir in include_dirs]
         cc_cmd += [f"/Fo{os.path.join(os.path.dirname(out), 'main.obj')}"]
         cc_cmd += ["/link"]
@@ -66,19 +66,22 @@ def _build(name, src, srcdir, library_dirs, include_dirs, libraries, extra_compi
             clangpp = shutil.which("clang++")
             gxx = shutil.which("g++")
             icpx = shutil.which("icpx")
-            cxx = icpx if os.name == "nt" else icpx or clangpp or gxx
+            cl = shutil.which("cl")
+            cxx = icpx or cl if os.name == "nt" else icpx or clangpp or gxx
             if cxx is None:
                 raise RuntimeError("Failed to find C++ compiler. Please specify via CXX environment variable.")
         cc = cxx
         import numpy as np
         numpy_include_dir = np.get_include()
         include_dirs = include_dirs + [numpy_include_dir]
-        if icpx is not None:
+        if cxx is icpx:
             extra_compile_args += ["-fsycl"]
         else:
             extra_compile_args += ["--std=c++17"]
         if os.name == "nt":
-            library_dirs = library_dirs + [os.path.join(sysconfig.get_paths(scheme=scheme)["stdlib"], "..", "libs")]
+            library_dirs = library_dirs + [
+                os.path.abspath(os.path.join(sysconfig.get_paths(scheme=scheme)["stdlib"], "..", "libs"))
+            ]
     else:
         cc_cmd = [cc]
 

third_party/intel/backend/driver.py

@@ -60,9 +60,12 @@ def find_sycl(include_dir: list[str]) -> tuple[list[str], str]:
         # sycl/sycl.hpp and sycl/CL/sycl.hpp results in both folders
         # being add: include and include/sycl.
         if f.name == "sycl.hpp":
-            include_dir += [f.locate().parent.parent.resolve().as_posix()]
-        if f.name == "libsycl.so":
-            sycl_dir = f.locate().parent.resolve().as_posix()
+            include_dir += [str(f.locate().parent.parent.resolve())]
+        if f.name in ["libsycl.so", "sycl8.dll"]:
+            sycl_dir = str(f.locate().parent.resolve())
+            # should we handle `_` somehow?
+            if os.name == "nt":
+                _ = os.add_dll_directory(sycl_dir)
 
     return include_dir, sycl_dir
 
@@ -76,9 +79,7 @@ def __init__(self):
         self._library_dir = None
         self._include_dir = None
         self._libsycl_dir = None
-        self.libraries = ['ze_loader']
-        if os.name != "nt":
-            self.libraries += ["sycl"]
+        self.libraries = ['ze_loader', 'sycl']
 
     @property
     def inject_pytorch_dep(self):
@@ -145,7 +146,11 @@ def compile_module_from_src(src, name):
                 f.write(src)
             extra_compiler_args = []
             if COMPILATION_HELPER.libsycl_dir:
-                extra_compiler_args += ['-Wl,-rpath,' + COMPILATION_HELPER.libsycl_dir]
+                if os.name == "nt":
+                    extra_compiler_args += ["/LIBPATH:" + COMPILATION_HELPER.libsycl_dir]
+                else:
+                    extra_compiler_args += ["-Wl,-rpath," + COMPILATION_HELPER.libsycl_dir]
+
             so = _build(name, src_path, tmpdir, COMPILATION_HELPER.library_dir, COMPILATION_HELPER.include_dir,
                         COMPILATION_HELPER.libraries, extra_compile_args=extra_compiler_args)
             with open(so, "rb") as f: