add docker run arguments for cuda

GEOS-DEV · May 23, 2024 · d309f39 · d309f39
1 parent 718d916
commit d309f39
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/continuousIntegration.yml b/.github/workflows/continuousIntegration.yml
@@ -140,7 +140,7 @@ jobs:
             CMAKE_C_COMPILER: /usr/bin/gcc
             CMAKE_BUILD_TYPE: Release
             ENABLE_CUDA: ON
-#            BUILD_AND_TEST_ARGS: "--disable-unit-tests"
+            DOCKER_RUN_ARGS: "--cpus=1 --memory=64g --runtime=nvidia --gpus all"
 
           - name: ubuntu22-gcc11-cuda11-dbg (ubuntu-22, gcc 11.4.0, cuda-11.8.0, Debug)
             DOCKER_REPOSITORY: geosx/ubuntu:22.04-cuda11.8
@@ -205,6 +205,7 @@ jobs:
         CMAKE_BUILD_TYPE: ${{ matrix.CMAKE_BUILD_TYPE }}
         ENABLE_CUDA: ${{ matrix.ENABLE_CUDA }}
         BUILD_AND_TEST_ARGS: ${{ matrix.BUILD_AND_TEST_ARGS }} --build-exe
+        DOCKER_RUN_ARGS: ${{ matrix.DOCKER_RUN_ARGS }}
         HOST_CONFIG: hostconfigs/environment.cmake
       run: ./scripts/ci_build_and_test.sh
 

diff --git a/scripts/ci_build_and_test.sh b/scripts/ci_build_and_test.sh
@@ -15,6 +15,7 @@ docker run \
   --name=${CONTAINER_NAME} \
   --volume=${BUILD_DIR}:${BUILD_DIR_MOUNT_POINT} \
   --cap-add=ALL \
+  ${DOCKER_RUN_ARGS} \
   -e HOST_CONFIG=${HOST_CONFIG} \
   -e CMAKE_C_COMPILER=${CMAKE_C_COMPILER} \
   -e CMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} \