From 7724722770ac7b3e2bafbcf5d619e62da7ecd24b Mon Sep 17 00:00:00 2001 From: Conner Swann <2635475+yourbuddyconner@users.noreply.github.com> Date: Tue, 17 Sep 2024 11:08:37 -0700 Subject: [PATCH] feat: added cuda action (#1492) --- .github/workflows/main.yml | 34 ++++++++++++++++++++++++++++++++++ crates/cuda/src/lib.rs | 1 - 2 files changed, 34 insertions(+), 1 deletion(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 72abf946de..f7b091c40f 100644 --- 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 \ No newline at end of file diff --git a/crates/cuda/src/lib.rs b/crates/cuda/src/lib.rs index ecaf627cba..66fb531816 100644 --- 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",