fix tests and update timeout time

MolloiLab · Sep 13, 2024 · 9ef144a · 9ef144a
1 parent 673c244
commit 9ef144a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/.buildkite/runbenchmarks.yml b/.buildkite/runbenchmarks.yml
@@ -26,7 +26,7 @@ steps:
           queue: "benchmark"
         env:
           BENCHMARK_GROUP: CPU
-        timeout_in_minutes: 5
+        timeout_in_minutes: 12
 
       - label: "CUDA: Run Benchmarks"
         key: "cuda-benchmarks"
@@ -48,7 +48,7 @@ steps:
           cuda: "*"
         env:
           BENCHMARK_GROUP: CUDA
-        timeout_in_minutes: 5
+        timeout_in_minutes: 12
 
       - label: "AMDGPU: Run Benchmarks"
         key: "amdgpu-benchmarks"
@@ -70,7 +70,7 @@ steps:
           rocm: "*"
         env:
           BENCHMARK_GROUP: AMDGPU
-        timeout_in_minutes: 5
+        timeout_in_minutes: 12
 
       - label: "oneAPI: Run Benchmarks"
         key: "oneapi-benchmarks"
@@ -92,7 +92,7 @@ steps:
           intel: "*"
         env:
           BENCHMARK_GROUP: oneAPI
-        timeout_in_minutes: 5
+        timeout_in_minutes: 12
 
       - label: "Metal: Run Benchmarks"
         key: "metal-benchmarks"
@@ -115,7 +115,7 @@ steps:
           arch: "arm64"
         env:
           BENCHMARK_GROUP: Metal
-        timeout_in_minutes: 5
+        timeout_in_minutes: 12
 
       - wait: ~  # Wait for all benchmarks to finish
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,5 +1,7 @@
 using DistanceTransforms
 using Test
+using KernelAbstractions
+using Random
 
 AVAILABLE_GPU_BACKENDS = ["CUDA", "AMDGPU", "Metal", "oneAPI"]
 TEST_BACKENDS = filter(x->x in [AVAILABLE_GPU_BACKENDS; "CPU"], ARGS)
@@ -40,8 +42,5 @@ else
     dev = Array
 end
 
-using KernelAbstractions
-using Random
-
 include("transform.jl")
 include("utils.jl")