set devcontainer to cuda12.2

benfred · Apr 8, 2024 · 1eb0f7f · 1eb0f7f
1 parent 8cc46d5
commit 1eb0f7f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
@@ -101,6 +101,8 @@ jobs:
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@branch-24.06
     with:
+      arch: '["amd64"]'
+      cuda: '["12.2"]'
       build_command: |
         sccache -z;
         build-all -DBUILD_PRIMS_BENCH=ON -DBUILD_ANN_BENCH=ON --verbose;