From 6779bac9b1e3e129e0df23e1a998e3344df0596e Mon Sep 17 00:00:00 2001
From: Saveliy Borisov <94908850+Amazingkivas@users.noreply.github.com>
Date: Thu, 7 Nov 2024 13:41:08 +0300
Subject: [PATCH] Update main.yml

---
 .github/workflows/main.yml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index ad758cd..8dd468e 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -66,12 +66,18 @@ jobs:
           sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-10 100
           echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
           source ~/.bashrc
+          NVCC_PATH=$(which nvcc)
+          echo "The path to nvcc is: $NVCC_PATH"
+          if [ -z "$NVCC_PATH" ]; then
+            echo "nvcc not found. Please check your CUDA installation."
+            exit 1
+          fi
           cd 3rdparty/kokkos
           mkdir build
           cd build
           cmake .. \
               -DCMAKE_CXX_COMPILER=${PWD}/../bin/nvcc_wrapper \
-              -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
+              -DCMAKE_CUDA_COMPILER=/usr/bin/nvcc \
               -DCMAKE_CUDA_HOST_COMPILER=/usr/bin/g++-10 \
               -DKokkos_ENABLE_CUDA=ON \
               -DKokkos_ARCH_AMPERE80=ON \