diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index d0826772271c3dddde4ddbcf7f1a0696db2c38a9..f71963fa78bbeef660cdce9aad1c6e05754059eb 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -230,6 +230,22 @@ gcc_9_python_bindings_test: ############################################################################### ## Benchmark ## ############################################################################### +nvidia_test: + stage: benchmark + + needs: [] + + tags: + - gpu + - linux + + image: "nvidia/cuda:11.1.1-devel-ubuntu16.04" + + script: + - echo NVIDIA_VISIBLE_DEVICES=${NVIDIA_VISIBLE_DEVICES} + - nvidia-smi + + gpu_numerical_tests: stage: benchmark