liger-kernel-nightly 0.5.9.dev20250519035525__tar.gz → 0.5.9.dev20250521172623__tar.gz

{liger_kernel_nightly-0.5.9.dev20250519035525 → liger_kernel_nightly-0.5.9.dev20250521172623}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.9.dev20250519035525
+Version: 0.5.9.dev20250521172623
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 

{liger_kernel_nightly-0.5.9.dev20250519035525 → liger_kernel_nightly-0.5.9.dev20250521172623}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.5.9.dev20250519035525"
+version = "0.5.9.dev20250521172623"
 description = "Efficient Triton kernels for LLM Training"
 urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
 readme = { file = "README.md", content-type = "text/markdown" }

{liger_kernel_nightly-0.5.9.dev20250519035525 → liger_kernel_nightly-0.5.9.dev20250521172623}/setup.py

@@ -47,6 +47,27 @@ def get_optional_dependencies():
     }
 
 
+def is_xpu_available():
+    """
+    Check if Intel XPU is available.
+    xpu-smi is often missing right now.
+    """
+    try:
+        subprocess.run(["xpu-smi"], check=True)
+        return True
+    except (subprocess.SubprocessError, FileNotFoundError):
+        pass
+
+    try:
+        result = subprocess.run("sycl-ls", check=True, capture_output=True, shell=True)
+        if 'level_zero:gpu' in result.stdout.decode():
+            return True
+    except (subprocess.SubprocessError, FileNotFoundError):
+        pass
+
+    return False
+
+
 def get_platform() -> Literal["cuda", "rocm", "cpu", "xpu"]:
     """
     Detect whether the system has NVIDIA or AMD GPU without torch dependency.
@@ -63,11 +84,10 @@ def get_platform() -> Literal["cuda", "rocm", "cpu", "xpu"]:
             print("ROCm GPU detected")
             return "rocm"
         except (subprocess.SubprocessError, FileNotFoundError):
-            try:
-                subprocess.run(["xpu-smi"], check=True)
+            if is_xpu_available():
                 print("Intel GPU detected")
                 return "xpu"
-            except (subprocess.SubprocessError, FileNotFoundError):
+            else:
                 print("No GPU detected")
                 return "cpu"
 

{liger_kernel_nightly-0.5.9.dev20250519035525 → liger_kernel_nightly-0.5.9.dev20250521172623}/src/liger_kernel_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.9.dev20250519035525
+Version: 0.5.9.dev20250521172623
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation