Skip to content

Commit

Permalink
Update main.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
Amazingkivas authored Nov 5, 2024
1 parent 4e31720 commit b02b573
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,8 @@ jobs:
run: |
sudo apt-get update
sudo apt-get install -y nvidia-cuda-toolkit
sudo apt-get install -y g++-10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-10 100
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
source ~/.bashrc
cd 3rdparty/kokkos
Expand All @@ -70,6 +72,7 @@ jobs:
cmake .. \
-DCMAKE_CXX_COMPILER=${PWD}/../bin/nvcc_wrapper \
-DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
-DCMAKE_CUDA_HOST_COMPILER=/usr/bin/g++-10 \
-DKokkos_ENABLE_CUDA=ON \
-DKokkos_ARCH_AMPERE80=ON \
-DCMAKE_CXX_STANDARD=17
Expand Down

0 comments on commit b02b573

Please sign in to comment.