Gpu test

.gitlab-ci.yml

@@ -1,10 +1,16 @@
-execute:
+gpu_test:
+  image: nvidia/cuda:11.5.1-devel-ubuntu18.04
+  tags:
+  - docker-gpu-nvidia
   script:
-    - echo "I succeed quickly"
-
-
-failing_job:
-  script:
-    - echo "I will also succeed now"
-    - sleep 15
-    - exit 0
+    - export PATH=/usr/local/sbin:/usr/sbin:/sbin:$PATH
+    - export PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
+    - apt update 
+    - apt-get -y install git
+    - git clone $CLONE_URL src
+    - cd src
+    - git checkout $HEAD_SHA
+    - cd ..
+    - ls src
+    - nvcc src/helloworld.cu -o helloworld
+    - ./helloworld

helloworld.cu

@@ -0,0 +1,99 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <math.h>
+#include <assert.h>
+#include <cuda.h>
+#include <cuda_runtime.h>
+#include <iostream>
+
+#define N 10000000
+#define MAX_ERR 1e-6
+
+__global__ void vector_add(float *out, float *a, float *b, int n) {
+    for(int i = 0; i < n; i ++){
+        out[i] = a[i] + b[i];
+    }
+}
+
+void display_infos()
+{
+    const int kb = 1024;
+    const int mb = kb * kb;
+    std::cout << "NBody.GPU" << std::endl << "=========" << std::endl << std::endl;
+
+    std::cout << "CUDA version:   v" << CUDART_VERSION << std::endl;
+
+    int devCount = 0;
+    cudaGetDeviceCount(&devCount);
+    std::cout << "Found " << devCount << " CUDA Devices: " << std::endl << std::endl;
+
+    for(int i = 0; i < devCount; ++i)
+    {
+        cudaDeviceProp props;
+        cudaGetDeviceProperties(&props, i);
+        std::cout << i << ": " << props.name << ": " << props.major << "." << props.minor << std::endl;
+        std::cout << "  Global memory:   " << props.totalGlobalMem / mb << "mb" << std::endl;
+        std::cout << "  Shared memory:   " << props.sharedMemPerBlock / kb << "kb" << std::endl;
+        std::cout << "  Constant memory: " << props.totalConstMem / kb << "kb" << std::endl;
+        std::cout << "  Block registers: " << props.regsPerBlock << std::endl << std::endl;
+
+        std::cout << "  Warp size:         " << props.warpSize << std::endl;
+        std::cout << "  Threads per block: " << props.maxThreadsPerBlock << std::endl;
+        std::cout << "  Max block dimensions: [ " << props.maxThreadsDim[0] << ", " << props.maxThreadsDim[1]  << ", " << props.maxThreadsDim[2] << " ]" << std::endl;
+        std::cout << "  Max grid dimensions:  [ " << props.maxGridSize[0] << ", " << props.maxGridSize[1]  << ", " << props.maxGridSize[2] << " ]" << std::endl;
+        std::cout << std::endl;
+    }
+}
+
+int main(){
+    display_infos();
+
+    float *a, *b, *out;
+    float *d_a, *d_b, *d_out; 
+
+    // Allocate host memory
+    a   = (float*)malloc(sizeof(float) * N);
+    b   = (float*)malloc(sizeof(float) * N);
+    out = (float*)malloc(sizeof(float) * N);
+
+    // Initialize host arrays
+    for(int i = 0; i < N; i++){
+        a[i] = 1.0f;
+        b[i] = 2.0f;
+    }
+
+    // Allocate device memory
+    cudaMalloc((void**)&d_a, sizeof(float) * N);
+    cudaMalloc((void**)&d_b, sizeof(float) * N);
+    cudaMalloc((void**)&d_out, sizeof(float) * N);
+
+    // Transfer data from host to device memory
+    cudaMemcpy(d_a, a, sizeof(float) * N, cudaMemcpyHostToDevice);
+    cudaMemcpy(d_b, b, sizeof(float) * N, cudaMemcpyHostToDevice);
+
+    // Executing kernel 
+    vector_add<<<1,1>>>(d_out, d_a, d_b, N);
+
+    // Transfer data back to host memory
+    cudaMemcpy(out, d_out, sizeof(float) * N, cudaMemcpyDeviceToHost);
+
+    // Verification
+    for(int i = 0; i < N; i++){
+        if(!(fabs(out[i] - a[i] - b[i]) < MAX_ERR)) {
+            printf("ERROR in verification");
+            return 0;
+        }
+    }
+    printf("out[0] = %f\n", out[0]);
+    printf("PASSED\n");
+
+    // Deallocate device memory
+    cudaFree(d_a);
+    cudaFree(d_b);
+    cudaFree(d_out);
+
+    // Deallocate host memory
+    free(a); 
+    free(b); 
+    free(out);
+}