feat: added cuda action (#1492)

succinctlabs · Sep 17, 2024 · 7724722 · 7724722
1 parent dff8b26
commit 7724722
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -167,3 +167,37 @@ jobs:
           cargo remove sp1-sdk
           cargo add sp1-sdk --path $GITHUB_WORKSPACE/crates/sdk
           SP1_DEV=1 RUST_LOG=info cargo run --release
+  test-cuda:
+    name: Test CUDA
+    runs-on: nvidia-gpu-linux-x64
+    steps:
+      - name: Checkout sources
+        uses: actions/checkout@v4
+
+      - name: rust-cache
+        uses: actions/cache@v3
+        with:
+          path: |
+            ~/.cargo/bin/
+            ~/.cargo/registry/index/
+            ~/.cargo/registry/cache/
+            ~/.cargo/git/db/
+            target/
+            ~/.rustup/
+          key: rust-1.79.0-${{ hashFiles('**/Cargo.toml') }}
+          restore-keys: rust-1.79.0-
+
+      - name: Setup toolchain
+        id: rustc-toolchain
+        shell: bash
+        run: |
+          curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- --default-toolchain 1.79.0 -y
+
+      - name: Run script
+        run: |
+          . "$HOME/.cargo/env"
+          curl -L https://sp1.succinct.xyz | bash
+          /home/runner/.sp1/bin/sp1up
+          sudo apt install libssl-dev pkg-config
+          cd examples/fibonacci
+          RUST_LOG=info cargo run --release --features cuda
diff --git a/crates/cuda/src/lib.rs b/crates/cuda/src/lib.rs
@@ -109,7 +109,6 @@ impl SP1CudaProver {
                 "-p",
                 "3000:3000",
                 "--rm",
-                "--runtime=nvidia",
                 "--gpus",
                 "all",
                 "--name",