liger-kernel-nightly 0.6.0.dev20250722224228__tar.gz → 0.6.0.dev20250722231330__tar.gz

{liger_kernel_nightly-0.6.0.dev20250722224228 → liger_kernel_nightly-0.6.0.dev20250722231330}/.github/workflows/benchmark.yml

@@ -1,6 +1,8 @@
 name: Benchmarks
 
 on:
+  release:
+    types: [published]
   schedule:
     # Runs at 00:00 UTC every Friday
     - cron: '0 0 * * 5'
@@ -69,10 +71,8 @@ jobs:
 
           # Save benchmark folder from main
           cp -r benchmark /tmp/benchmark_main
-
           # Checkout back to target commit
           git checkout $ORIG_COMMIT
-
           # Replace old benchmark with one from main
           rm -rf benchmark
           cp -r /tmp/benchmark_main benchmark
@@ -124,8 +124,17 @@ jobs:
       
       # Step 6: Copy benchmark CSV to gh-pages directory
       - name: Copy generated benchmark to gh-pages
+        id: copy_benchmark
         run: |
-          COMMIT_DIR="gh-pages/${OUTPUT_DIR}/${{ steps.choose_commit.outputs.hash }}"
+          if [[ "${{ github.event_name }}" == "release" ]]; then
+            echo "Release event detected"
+            path=${{steps.choose_commit.outputs.hash}}-${{ github.event.release.tag_name }}
+          else
+            echo "Not a release event"
+            path=${{steps.choose_commit.outputs.hash}}
+          fi
+          COMMIT_DIR="gh-pages/${OUTPUT_DIR}/${path}"
+
           mkdir -p "$COMMIT_DIR"
 
           if [ -f "$COMMIT_DIR/${OUTPUT_FILENAME}" ]; then
@@ -144,7 +153,8 @@ jobs:
           mkdir -p ${OUTPUT_DIR}
           touch ${OUTPUT_DIR}/commits.txt
 
-          echo "${{ steps.choose_commit.outputs.hash }}" >> ${OUTPUT_DIR}/commits.txt
+          echo "${{ steps.copy_benchmark.outputs.path }}" >> ${OUTPUT_DIR}/commits.txt
+
           echo "Added commit hash to commits.txt"
       # Step 7: Commit and push
       - name: Commit and push to gh-pages
@@ -153,6 +163,5 @@ jobs:
           git config user.name github-actions[bot]
           git config user.email 41898282+github-actions[bot]@users.noreply.github.com
           git add .
-          git commit -m "Add benchmark for commit ${{ steps.choose_commit.outputs.hash }}" || echo "No changes to commit"
+          git commit -m "Add benchmark for commit ${{ steps.copy_benchmark.outputs.path }}" || echo "No changes to commit"
           git push origin gh-pages
-

{liger_kernel_nightly-0.6.0.dev20250722224228 → liger_kernel_nightly-0.6.0.dev20250722231330}/PKG-INFO

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

{liger_kernel_nightly-0.6.0.dev20250722224228 → liger_kernel_nightly-0.6.0.dev20250722231330}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.6.0.dev20250722224228"
+version = "0.6.0.dev20250722231330"
 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.6.0.dev20250722224228 → liger_kernel_nightly-0.6.0.dev20250722231330}/src/liger_kernel/ops/layer_norm.py

@@ -256,8 +256,8 @@ def layer_norm_backward(dY, X, W, B, Mean, RSTD):
     # Use float32 for atomic operations if bfloat16 is not supported
     atomic_dtype = tl.float32 if triton_dtype == tl.bfloat16 else triton_dtype
 
+    kernel_args = {"num_warps": num_warps}
     # XPU-specific optimization
-    kernel_args = {}
     if X.device.type == "xpu":
         kernel_args.update({"grf_mode": "large", "num_warps": 32, "num_stages": 4})
 
@@ -279,7 +279,6 @@ def layer_norm_backward(dY, X, W, B, Mean, RSTD):
         BLOCK_SIZE=BLOCK_SIZE,
         dtype=triton_dtype,
         atomic_dtype=atomic_dtype,
-        num_warps=num_warps,
         **kernel_args,
     )
 

{liger_kernel_nightly-0.6.0.dev20250722224228 → liger_kernel_nightly-0.6.0.dev20250722231330}/src/liger_kernel_nightly.egg-info/PKG-INFO

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