liger-kernel-nightly 0.5.3.dev20250210231423__tar.gz → 0.5.3.dev20250211214737__tar.gz

liger_kernel_nightly-0.5.3.dev20250211214737/.github/workflows/intel-ci.yml

@@ -0,0 +1,71 @@
+name: Intel GPU
+
+on:
+  push:
+    branches:
+      - main
+    paths:
+      - "src/**"
+      - "test/**"
+  pull_request:
+    branches:
+      - main
+    paths:
+      - "src/**"
+      - "test/**"
+  schedule:
+    # Runs at 00:00 UTC daily
+    - cron: '0 0 * * *'
+  workflow_dispatch:  # Enables manual trigger
+
+concurrency:
+  # This causes it to cancel previous in-progress actions on the same PR / branch,
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  checkstyle:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v3
+      with:
+        python-version: '3.10'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r dev/fmt-requirements.txt
+
+    - name: Run checkstyle
+      run: make checkstyle
+
+  tests:
+    runs-on: linux-max1550-gpu-8
+    needs: [checkstyle]
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v3
+      with:
+        python-version: '3.10'
+
+    - name: Setup Dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -e .[dev] --extra-index-url https://download.pytorch.org/whl/test/xpu
+    
+    - name: List Python Environments
+      run: python -m pip list
+    
+    - name: Run Unit Tests
+      run: |
+        make test
+        make test-convergence

{liger_kernel_nightly-0.5.3.dev20250210231423 → liger_kernel_nightly-0.5.3.dev20250211214737}/PKG-INFO

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

{liger_kernel_nightly-0.5.3.dev20250210231423 → liger_kernel_nightly-0.5.3.dev20250211214737}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.5.3.dev20250210231423"
+version = "0.5.3.dev20250211214737"
 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.3.dev20250210231423 → liger_kernel_nightly-0.5.3.dev20250211214737}/setup.py

@@ -21,6 +21,11 @@ def get_default_dependencies():
             "torch>=2.6.0.dev",
             "triton>=3.0.0",
         ]
+    elif platform == "xpu":
+        return [
+            "torch>=2.6.0",
+            "pytorch-triton-xpu>=3.2.0",
+        ]
 
 
 def get_optional_dependencies():
@@ -43,8 +48,7 @@ def get_optional_dependencies():
     }
 
 
-# TODO: add intel XPU
-def get_platform() -> Literal["cuda", "rocm", "cpu"]:
+def get_platform() -> Literal["cuda", "rocm", "cpu", "xpu"]:
     """
     Detect whether the system has NVIDIA or AMD GPU without torch dependency.
     """
@@ -60,8 +64,13 @@ def get_platform() -> Literal["cuda", "rocm", "cpu"]:
             print("ROCm GPU detected")
             return "rocm"
         except (subprocess.SubprocessError, FileNotFoundError):
-            print("No GPU detected")
-            return "cpu"
+            try:
+                subprocess.run(["xpu-smi"], check=True)
+                print("Intel GPU detected")
+                return "xpu"
+            except (subprocess.SubprocessError, FileNotFoundError):
+                print("No GPU detected")
+                return "cpu"
 
 
 setup(

{liger_kernel_nightly-0.5.3.dev20250210231423 → liger_kernel_nightly-0.5.3.dev20250211214737}/src/liger_kernel_nightly.egg-info/PKG-INFO

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

{liger_kernel_nightly-0.5.3.dev20250210231423 → liger_kernel_nightly-0.5.3.dev20250211214737}/src/liger_kernel_nightly.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ setup.py
 .github/ISSUE_TEMPLATE/feature_request.yaml
 .github/workflows/amd-ci.yml
 .github/workflows/docs.yml
+.github/workflows/intel-ci.yml
 .github/workflows/nvi-ci.yml
 .github/workflows/publish-nightly.yml
 .github/workflows/publish-release.yml