Codspeed setup

EmbeddedLLM · Dec 18, 2023 · 08508b3 · 08508b3
1 parent d3307bc
commit 08508b3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/.github/workflows/codspeed.yml b/.github/workflows/codspeed.yml
@@ -26,11 +26,20 @@ jobs:
         with:
           python-version: "3.10"
 
+      - name: cuda-toolkit
+        uses: Jimver/[email protected]
+        with:
+          cuda: "12.3.0"
+
+      - run: echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+      - run: echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+      - run: nvcc -V  
+
       - name: Install project dependencies
         run: pip install -e . && pip install pytest
 
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v2
         with:
           token: ${{ secrets.codspeed_token }}
-          run: "pytest tests/kernels/test_layernorm.py --codspeed"
+          run: "pytest tests/kernels/test_layernorm.py --codspeed"