Merge branch 'hf/28-compilation-on-orin' of github.com:GPUEngineering/GPUtils into hf/28-compilation-on-orin

Author: Ruairi Moran

.github/workflows/ci.yml

@@ -7,7 +7,10 @@ name: Continuous integration
 
 jobs:
   ci:
-    runs-on: self-hosted
+    runs-on: ${{ matrix.runner }}
+    strategy:
+      matrix:
+        runner: [orin, a40]
     steps:
       - name: checkout code
         uses: actions/checkout@v4

ci/script.sh

@@ -1,44 +1,62 @@
 #!/bin/bash
 set -euxo pipefail
 
+
 tests() {
+    # Where are we? (A40 or Orin?)
+    cpp_version=17 # default
+    sm_arch=86 # default
+    hwInfoOrin=`lshw | grep Orin` ||
+    if [ ! -z "${hwInfoOrin}" ]; then
+      echo "Running on Orin";
+      sm_arch=87
+      cpp_version=17
+    else
+      echo "Not running on Orin";
+      sm_arch=86
+      cpp_version=20
+    fi
+
     # ------------------------------------
     # Run tensor gtests
     # ------------------------------------
 
     # -- create build files
-    cmake -S . -B ./build -Wno-dev
+    cmake -DCPPVERSION=${cpp_version} -DSM_ARCH=${sm_arch} -S . -B ./build -Wno-dev
 
     # -- build files in build folder
     cmake --build ./build
 
     # -- run tests
     ctest --test-dir ./build/test --output-on-failure
 
-    # -- run compute sanitizer
-    cd ./build/test
-    mem=$(/usr/local/cuda-12.3/bin/compute-sanitizer --tool memcheck --leak-check=full ./device_test)
-    grep "0 errors" <<< "$mem"
-    cd ../..
+    if [ -z "${hwInfoOrin}" ]; then
 
-    # ------------------------------------
-    # Run example executable
-    # ------------------------------------
+      # -- run compute sanitizer
+      cd ./build/test
+      mem=$(/usr/local/cuda/bin/compute-sanitizer --tool memcheck --leak-check=full ./device_test)
+      grep "0 errors" <<< "$mem"
+      cd ../..
 
-    # -- create build files
-    cd example
-    cmake -S . -B ./build -Wno-dev
+      # ------------------------------------
+      # Run example executable
+      # ------------------------------------
 
-    # -- build files in build folder
-    cmake --build ./build
+      # -- create build files
+      cd example
+      cmake  -DCPPVERSION=${cpp_version} -DSM_ARCH=${sm_arch} -S . -B ./build -Wno-dev
+
+      # -- build files in build folder
+      cmake --build ./build
 
-    # -- run main.cu
-    ./build/example_main
+      # -- run main.cu
+      ./build/example_main
 
-    # -- run compute sanitizer
-    cd ./build
-    mem=$(/usr/local/cuda-12.3/bin/compute-sanitizer --tool memcheck --leak-check=full ./example_main)
-    grep "0 errors" <<< "$mem"
+      # -- run compute sanitizer
+      cd ./build
+      mem=$(/usr/local/cuda/bin/compute-sanitizer --tool memcheck --leak-check=full ./example_main)
+      grep "0 errors" <<< "$mem"
+    fi
 }