Debuggin of triton error

facebookresearch · Oct 30, 2024 · f91825c · f91825c
1 parent 492d9bf
commit f91825c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/gpu-tests.yaml b/.github/workflows/gpu-tests.yaml
@@ -19,6 +19,10 @@ jobs:
           run: |
             uv venv && source .venv/bin/activate
             uv pip install -e ".[dev]"
+        - name: Check Triton installation.
+          run: |
+            source .venv/bin/activate
+            python3 -c "import triton; print(triton.__version__)"
         - name: Run tests.
           run: |
             source .venv/bin/activate