triton-windows 3.3.0a0.post11__cp310-cp310-win_amd64.whl → 3.3.0a0.post12__cp310-cp310-win_amd64.whl

triton/backends/nvidia/driver.py

@@ -23,14 +23,14 @@ libraries = ['cuda']
 
 @functools.lru_cache()
 def libcuda_dirs():
-    if os.name == "nt":
-        _, _, cuda_lib_dirs = find_cuda()
-        return cuda_lib_dirs
-
     env_libcuda_path = os.getenv("TRITON_LIBCUDA_PATH")
     if env_libcuda_path:
         return [env_libcuda_path]
 
+    if os.name == "nt":
+        _, _, cuda_lib_dirs = find_cuda()
+        return cuda_lib_dirs
+
     libs = subprocess.check_output(["/sbin/ldconfig", "-p"]).decode()
     # each line looks like the following:
     # libcuda.so.1 (libc6,x86-64) => /lib/x86_64-linux-gnu/libcuda.so.1

triton/windows_utils.py

@@ -4,6 +4,7 @@ import re
 import subprocess
 import sys
 import sysconfig
+import warnings
 import winreg
 from collections.abc import Iterable
 from functools import partial
@@ -150,7 +151,7 @@ def find_msvc() -> tuple[list[str], list[str]]:
     if msvc_base_path is None:
         msvc_base_path, version = find_msvc_hardcoded()
     if msvc_base_path is None:
-        print("WARNING: Failed to find MSVC.")
+        warnings.warn("Failed to find MSVC.")
         return [], []
 
     return (
@@ -216,7 +217,7 @@ def find_winsdk() -> tuple[list[str], list[str]]:
     if winsdk_base_path is None:
         winsdk_base_path, version = find_winsdk_hardcoded()
     if winsdk_base_path is None:
-        print("WARNING: Failed to find Windows SDK.")
+        warnings.warn("Failed to find Windows SDK.")
         return [], []
 
     return (
@@ -250,126 +251,123 @@ def find_python() -> list[str]:
         if (python_lib_dir / "python3.lib").exists():
             return [str(python_lib_dir)]
 
-    print("WARNING: Failed to find Python libs.")
+    warnings.warn("Failed to find Python libs.")
     return []
 
 
-def find_cuda_bundled() -> tuple[Optional[str], list[str], list[str]]:
-    cuda_base_path = (
-        Path(sysconfig.get_paths()["platlib"]) / "triton" / "backends" / "nvidia"
-    )
-    if check_cuda_system_wide(cuda_base_path):
+def check_and_find_cuda(base_path: Path) -> tuple[Optional[str], list[str], list[str]]:
+    # pip
+    if all(
+        x.exists()
+        for x in [
+            base_path / "cuda_nvcc" / "bin" / "ptxas.exe",
+            base_path / "cuda_runtime" / "include" / "cuda.h",
+            base_path / "cuda_runtime" / "lib" / "x64" / "cuda.lib",
+        ]
+    ):
         return (
-            str(cuda_base_path / "bin"),
-            [str(cuda_base_path / "include")],
-            [str(cuda_base_path / "lib" / "x64")],
+            str(base_path / "cuda_nvcc" / "bin"),
+            [str(base_path / "cuda_runtime" / "include")],
+            [str(base_path / "cuda_runtime" / "lib" / "x64")],
         )
 
-    return None, [], []
-
-
-def check_cuda_pip(nvidia_base_path: Path) -> bool:
-    return all(
+    # conda
+    if all(
         x.exists()
         for x in [
-            nvidia_base_path / "cuda_nvcc" / "bin" / "ptxas.exe",
-            nvidia_base_path / "cuda_runtime" / "include" / "cuda.h",
-            nvidia_base_path / "cuda_runtime" / "lib" / "x64" / "cuda.lib",
+            base_path / "bin" / "ptxas.exe",
+            base_path / "include" / "cuda.h",
+            base_path / "lib" / "cuda.lib",
         ]
-    )
-
-
-def find_cuda_pip() -> tuple[Optional[str], list[str], list[str]]:
-    nvidia_base_path = Path(sysconfig.get_paths()["platlib"]) / "nvidia"
-    if check_cuda_pip(nvidia_base_path):
+    ):
         return (
-            str(nvidia_base_path / "cuda_nvcc" / "bin"),
-            [str(nvidia_base_path / "cuda_runtime" / "include")],
-            [str(nvidia_base_path / "cuda_runtime" / "lib" / "x64")],
+            str(base_path / "bin"),
+            [str(base_path / "include")],
+            [str(base_path / "lib")],
         )
 
-    return None, [], []
-
-
-def check_cuda_conda(cuda_base_path: Path) -> bool:
-    return all(
+    # bundled or system-wide
+    if all(
         x.exists()
         for x in [
-            cuda_base_path / "bin" / "ptxas.exe",
-            cuda_base_path / "include" / "cuda.h",
-            cuda_base_path / "lib" / "cuda.lib",
+            base_path / "bin" / "ptxas.exe",
+            base_path / "include" / "cuda.h",
+            base_path / "lib" / "x64" / "cuda.lib",
         ]
-    )
-
-
-def find_cuda_conda() -> tuple[Optional[str], list[str], list[str]]:
-    cuda_base_path = Path(sys.exec_prefix) / "Library"
-    if check_cuda_conda(cuda_base_path):
+    ):
         return (
-            str(cuda_base_path / "bin"),
-            [str(cuda_base_path / "include")],
-            [str(cuda_base_path / "lib")],
+            str(base_path / "bin"),
+            [str(base_path / "include")],
+            [str(base_path / "lib" / "x64")],
         )
 
     return None, [], []
 
 
-def check_cuda_system_wide(cuda_base_path: Path) -> bool:
-    return all(
-        x.exists()
-        for x in [
-            cuda_base_path / "bin" / "ptxas.exe",
-            cuda_base_path / "include" / "cuda.h",
-            cuda_base_path / "lib" / "x64" / "cuda.lib",
-        ]
-    )
-
-
-def find_cuda_env() -> Optional[Path]:
+def find_cuda_env() -> tuple[Optional[str], list[str], list[str]]:
     for cuda_base_path in ["CUDA_PATH", "CUDA_HOME"]:
         cuda_base_path = os.getenv(cuda_base_path)
         if cuda_base_path is None:
             continue
 
         cuda_base_path = Path(cuda_base_path)
-        if check_cuda_system_wide(cuda_base_path):
-            return cuda_base_path
+        cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs = check_and_find_cuda(
+            cuda_base_path
+        )
+        if cuda_bin_path:
+            return cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs
+
+    return None, [], []
 
-    return None
 
+def find_cuda_bundled() -> tuple[Optional[str], list[str], list[str]]:
+    cuda_base_path = (
+        Path(sysconfig.get_paths()["platlib"]) / "triton" / "backends" / "nvidia"
+    )
+    return check_and_find_cuda(cuda_base_path)
+
+
+def find_cuda_pip() -> tuple[Optional[str], list[str], list[str]]:
+    nvidia_base_path = Path(sysconfig.get_paths()["platlib"]) / "nvidia"
+    return check_and_find_cuda(nvidia_base_path)
+
+
+def find_cuda_conda() -> tuple[Optional[str], list[str], list[str]]:
+    cuda_base_path = Path(sys.exec_prefix) / "Library"
+    return check_and_find_cuda(cuda_base_path)
 
-def find_cuda_hardcoded() -> Optional[Path]:
+
+def find_cuda_hardcoded() -> tuple[Optional[str], list[str], list[str]]:
     parent = find_in_program_files(r"NVIDIA GPU Computing Toolkit\CUDA")
     if parent is None:
-        return None
+        return None, [], []
 
     paths = glob(str(parent / "v12*"))
     # First try the highest version
     paths = sorted(paths)[::-1]
-    for path in paths:
-        cuda_base_path = Path(path)
-        if check_cuda_system_wide(cuda_base_path):
-            return cuda_base_path
+    for cuda_base_path in paths:
+        cuda_base_path = Path(cuda_base_path)
+        cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs = check_and_find_cuda(
+            cuda_base_path
+        )
+        if cuda_bin_path:
+            return cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs
 
-    return None
+    return None, [], []
 
 
 @functools.cache
 def find_cuda() -> tuple[Optional[str], list[str], list[str]]:
-    for f in [find_cuda_bundled, find_cuda_pip, find_cuda_conda]:
+    for f in [
+        find_cuda_env,
+        find_cuda_bundled,
+        find_cuda_pip,
+        find_cuda_conda,
+        find_cuda_hardcoded,
+    ]:
         cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs = f()
         if cuda_bin_path:
             return cuda_bin_path, cuda_inc_dirs, cuda_lib_dirs
 
-    cuda_base_path = find_cuda_env()
-    if cuda_base_path is None:
-        cuda_base_path = find_cuda_hardcoded()
-    if cuda_base_path is None:
-        print("WARNING: Failed to find CUDA.")
-        return None, [], []
-
-    return (
-        str(cuda_base_path / "bin"),
-        [str(cuda_base_path / "include")],
-        [str(cuda_base_path / "lib" / "x64")],
-    )
+    warnings.warn("Failed to find CUDA.")
+    return None, [], []

{triton_windows-3.3.0a0.post11.dist-info → triton_windows-3.3.0a0.post12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: triton-windows
-Version: 3.3.0a0.post11
+Version: 3.3.0a0.post12
 Summary: A language and compiler for custom Deep Learning operations
 Home-page: https://github.com/woct0rdho/triton-windows
 Author: Philippe Tillet, Dian Wu

{triton_windows-3.3.0a0.post11.dist-info → triton_windows-3.3.0a0.post12.dist-info}/RECORD

@@ -3,8 +3,8 @@ triton/_internal_testing.py,sha256=OBY28huiEWItqGgiukgZzHLLaSbS8yj9kdhn_u562Yg,5
 triton/_utils.py,sha256=5RiCLwW14w0Q3mdZ-9yz-VO5KiSexNj9xeDt4gaNsvE,1014
 triton/errors.py,sha256=8WfnuRKLG578mgY6cBA3ECruVMf9ULEKFNgRcJ6IhWM,89
 triton/testing.py,sha256=ivFf1Fq9frmfVahaVUp0bgJxmvVZNACZfj3Sai6zfAs,20048
-triton/windows_utils.py,sha256=OStlF-dMd-rM7CbTknEp3A7q_iU8a5Ou4U-XjjfFwsI,10898
-triton/_C/libtriton.pyd,sha256=G6uGbfS8ASkqmjGuZdLJVD0010sojonUJSCO9lBhnu8,86790144
+triton/windows_utils.py,sha256=aQMItmuZNXaki8zSB7HMvhy1RsXlmz1GcO-lORm1IIk,10852
+triton/_C/libtriton.pyd,sha256=dSFneMmzTYzMqUaulmD-pIfiQnQAdYglKrvNZUDIsj8,86790144
 triton/backends/__init__.py,sha256=opAo_vgEMt3tLO_bYFrYGksnIu0qohbmyuu_s3-rNAs,1595
 triton/backends/compiler.py,sha256=ymaG0kpveAuESbQ9QZ0RyXjr0Aq4el_G5XGYogJ2gNA,3588
 triton/backends/driver.py,sha256=AN60upJlPgia0JwvZ8vIVgLITNPuI0fdz8zMIIHPpF4,1450
@@ -114,7 +114,7 @@ triton/backends/amd/lib/ocml.bc,sha256=UPNTXW0gCXUNB-c6orSYwb-mz9_mjUc7zny_vfFza
 triton/backends/nvidia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 triton/backends/nvidia/compiler.py,sha256=zxfzVLLdaH4HLldiM9bG3jXvf3w61gQ6Fkt1CQHQ7lU,18567
 triton/backends/nvidia/driver.c,sha256=-O-Q00SFcCHJiHvArIC7yB10Ma1W8jw7ORFKp7spxHE,18846
-triton/backends/nvidia/driver.py,sha256=EurNVIPoV9_McPtk729g1e0edW7lj4uXnQ19eMOKWmk,20843
+triton/backends/nvidia/driver.py,sha256=9ZhZTs19M5IU4fAD4JBNhh24MYcP6QLmEMW3SfGgfAQ,20843
 triton/backends/nvidia/bin/ptxas.exe,sha256=iCva9hIYg-Q2NybchwaQJFkwDzNiliFOTDdZrHPLb6A,24732160
 triton/backends/nvidia/include/cuda.h,sha256=Fn44OjeRImxegJ39apYUspseEfTWNGwpqSGUOnHj5WY,1183268
 triton/backends/nvidia/lib/libdevice.10.bc,sha256=XC-uN8huaMOjhgWpX1EtfRLV89uYYxC-R_VzBKpype4,473728
@@ -157,7 +157,7 @@ triton/tools/link.py,sha256=u7qtfZRLriZkAMEGNvj8YF-k1cthmLL7BwHYqBgT63E,11871
 triton/tools/mxfp.py,sha256=YQdpBrGkOVNOtnLeRjMCeVFHWkSwUubGeWsItIjO8TU,11737
 triton/tools/extra/cuda/compile.c,sha256=Me7beHPc6WNTwjg85H84DUMCRu4KJdVK2hNNgvlhBZ4,2126
 triton/tools/extra/cuda/compile.h,sha256=n9QKIFZTL4RSsiXtAxBP9XGSnxjyaevQQ9bBpwDsvAg,332
-triton_windows-3.3.0a0.post11.dist-info/METADATA,sha256=1DIjb6wwcr7nfqHpyU8hB-7JUlXkdT3OWK4-6a9-lls,1629
-triton_windows-3.3.0a0.post11.dist-info/WHEEL,sha256=H5gvOYEhGgVYizXMDNmO6CnZIviGx48geV3Lbzu7OZg,101
-triton_windows-3.3.0a0.post11.dist-info/top_level.txt,sha256=KhMzHYsArnZ3IkjAQ-xLnx1n_FjvEpJNelg2xPiDl-U,254
-triton_windows-3.3.0a0.post11.dist-info/RECORD,,
+triton_windows-3.3.0a0.post12.dist-info/METADATA,sha256=ZtZEEO2XubX1BJp1mHsG3DTUDMqt2R0z_rYh_K1QAFM,1629
+triton_windows-3.3.0a0.post12.dist-info/WHEEL,sha256=H5gvOYEhGgVYizXMDNmO6CnZIviGx48geV3Lbzu7OZg,101
+triton_windows-3.3.0a0.post12.dist-info/top_level.txt,sha256=KhMzHYsArnZ3IkjAQ-xLnx1n_FjvEpJNelg2xPiDl-U,254
+triton_windows-3.3.0a0.post12.dist-info/RECORD,,