Merge pull request #14 from aurianer/add_ci_on_gh200

Add CI on daint to also test with nvidia on GH200

Author: aurianer

.gitlab/includes/pipeline_daint.yml

@@ -0,0 +1,44 @@
+include:
+  - local: '.gitlab/includes/common_pipeline.yml'
+
+stages:
+  - build
+  - test
+
+variables:
+  UENV_IMAGE: "prgenv-gnu/25.6:v2"
+  WITH_UENV_VIEW: "default"
+  SLURM_PARTITION: "normal"
+
+
+build_benchmarks_gh200:
+  extends:
+    - .uenv-runner-daint-gh200
+    - .cmake_common
+    - .build_common
+  stage: build
+  image: $UENV_IMAGE
+  script:
+    - cmake --version
+    - nvcc --version || echo "CUDA not found in PATH"
+
+    - cmake -Bbuild $CMAKE_COMMON_FLAGS
+        -DWITH_HIP=OFF
+        -DWITH_CUDA=ON
+    - cmake --build build
+
+  artifacts:
+    paths:
+      - build/
+    expire_in: 1 hour
+
+
+test_benchmarks_gh200:
+  needs: [build_benchmarks_gh200]
+  extends:
+    - .uenv-runner-daint-gh200
+    - .test_common
+  stage: test
+  image: $UENV_IMAGE
+  script:
+    - ctest --test-dir build --output-on-failure

.gitlab/pipeline_prod.yml

@@ -0,0 +1,2 @@
+include:
+  - local: '.gitlab/includes/pipeline_daint.yml'

parallel_algos/CMakeLists.txt

@@ -18,6 +18,9 @@ option(WITH_HIP "Enable building for AMD GPUs" ON)
 if(WITH_CUDA)
     check_language(CUDA)
     if(CMAKE_CUDA_COMPILER)
+        if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+            set(CMAKE_CUDA_ARCHITECTURES 90)
+        endif()
         enable_language(CUDA)
         find_package(CUDAToolkit)
         set(CMAKE_CUDA_STANDARD 20)
@@ -30,6 +33,9 @@ endif()
 if(WITH_HIP)
     check_language(HIP)
     if(CMAKE_HIP_COMPILER)
+        if(NOT DEFINED CMAKE_HIP_ARCHITECTURES)
+            set(CMAKE_HIP_ARCHITECTURES gfx90a gfx942)
+        endif()
         enable_language(HIP)
         find_package(hip)
         set(CMAKE_HIP_STANDARD 20)