Skip to content

Commit

Permalink
Use gpu runner for build
Browse files Browse the repository at this point in the history
  • Loading branch information
popzxc committed Aug 16, 2024
1 parent baea64c commit 231d7a7
Showing 1 changed file with 1 addition and 17 deletions.
18 changes: 1 addition & 17 deletions .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,9 @@ on:
jobs:
build:
name: cargo build
runs-on: [ubuntu-latest]
strategy:
matrix:
cuda: [ "12.0.0-devel-ubuntu20.04", "12.5.0-devel-ubuntu20.04" ]
container:
image: nvidia/cuda:${{ matrix.cuda }}
runs-on: [ matterlabs-ci-gpu-runner ]
env:
BELLMAN_CUDA_DIR: ${{ github.workspace }}/bellman-cuda
CUDAARCHS: 89
steps:
- name: Prepare environment
env:
Expand All @@ -32,16 +26,6 @@ jobs:
echo "${HOME}/.cargo/bin" >> "${GITHUB_PATH}"
echo "export PATH=\"$HOME/.cargo/bin:\$PATH\"" >> "${HOME}/.bash_profile"
- name: Setup CMake
run: |
curl -LO https://github.com/Kitware/CMake/releases/download/v3.24.3/cmake-3.24.3-linux-x86_64.sh && \
chmod +x cmake-3.24.3-linux-x86_64.sh && \
./cmake-3.24.3-linux-x86_64.sh --skip-license --prefix=/usr/local
- name: Check CMake version
run: |
cmake --version
- name: Check CUDA version
run: |
nvcc --version
Expand Down

0 comments on commit 231d7a7

Please sign in to comment.