Add gpu env back in

Author: charleskawczynski

.buildkite/pipeline.yml

@@ -35,32 +35,33 @@ steps:
       queue: central
       slurm_ntasks: 1
 
-  # - label: "init gpu env"
-  #   key: "init_gpu_env"
-  #   command:
-  #     - echo "--- Configure MPI"
-  #     - julia -e 'using Pkg; Pkg.add("MPIPreferences"); using MPIPreferences; use_system_binary()'
+  - label: "init gpu env"
+    key: "init_gpu_env"
+    command:
+      - "echo $$JULIA_DEPOT_PATH"
+      - echo "--- Configure MPI"
+      - julia -e 'using Pkg; Pkg.add("MPIPreferences"); using MPIPreferences; use_system_binary()'
 
-  #     - echo "--- Instantiate project"
-  #     - "julia --project -e 'using Pkg; Pkg.instantiate(;verbose=true)'"
-  #     - "julia --project -e 'using Pkg; Pkg.precompile(;strict=true)'"
+      - echo "--- Instantiate project"
+      - "julia --project -e 'using Pkg; Pkg.instantiate(;verbose=true)'"
+      - "julia --project -e 'using Pkg; Pkg.precompile(;strict=true)'"
 
-  #     - echo "--- Instantiate test"
-  #     - "julia --project=test -e 'using Pkg; Pkg.develop(path=\".\")'"
-  #     - "julia --project=test -e 'using Pkg; Pkg.instantiate(;verbose=true)'"
-  #     - "julia --project=test -e 'using Pkg; Pkg.precompile()'"
+      - echo "--- Instantiate test"
+      - "julia --project=test -e 'using Pkg; Pkg.develop(path=\".\")'"
+      - "julia --project=test -e 'using Pkg; Pkg.instantiate(;verbose=true)'"
+      - "julia --project=test -e 'using Pkg; Pkg.precompile(;strict=true)'"
 
-  #     - echo "--- Initialize CUDA runtime"
-  #     - "julia --project -e 'using CUDA; CUDA.precompile_runtime()'"
-  #     - "julia --project -e 'using CUDA; CUDA.versioninfo()'"
+      - echo "--- Initialize CUDA runtime"
+      - "julia --project -e 'using CUDA; CUDA.precompile_runtime()'"
+      - "julia --project -e 'using CUDA; CUDA.versioninfo()'"
 
-  #     - echo "--- Package status"
-  #     - "julia --project -e 'using Pkg; Pkg.status()'"
-  #   agents:
-  #     config: gpu
-  #     queue: central
-  #     slurm_ntasks: 1
-  #     slurm_gres: "gpu:1"
+      - echo "--- Package status"
+      - "julia --project -e 'using Pkg; Pkg.status()'"
+    agents:
+      config: gpu
+      queue: central
+      slurm_ntasks: 1
+      slurm_gres: "gpu:1"
 
   - wait
 
@@ -115,3 +116,11 @@ steps:
       config: cpu
       queue: central
       slurm_ntasks: 1
+
+  - label: "GPU"
+    command:
+      - "julia --project=test test/runtests.jl CuArray"
+    agents:
+      slurm_gres: "gpu:1"
+      queue: central
+      slurm_ntasks: 1