feat: add flashinfer as 3rdparty and use rmsnorm as example (#3033)

2025-01-21 20:44:49 +08:00
parent a4331cd260
commit 5a0d680a14
11 changed files with 335 additions and 2 deletions
--- a/.github/workflows/pr-test-sgl-kernel.yml
+++ b/.github/workflows/pr-test-sgl-kernel.yml
@@ -41,6 +41,7 @@ jobs:
      - name: Install
        run: |
          pip3 install torch==2.5.1
+          pip3 install pytest
          pip3 uninstall sgl-kernel -y || true
          cd sgl-kernel
          pip3 install .