Widen gpu allocations buffer

CliMA · Apr 5, 2024 · 6c2fdfc · 6c2fdfc
1 parent b59fe2a
commit 6c2fdfc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/perf/flame.jl b/perf/flame.jl
@@ -88,8 +88,10 @@ using Test
 # Threaded/gpu allocations are not deterministic, so let's add a buffer
 # TODO: remove buffer, and threaded tests, when
 #       threaded/unthreaded functions are unified
-buffer = if any(x -> occursin(x, job_id), ("threaded", "gpu"))
+buffer = if any(x -> occursin(x, job_id), ("threaded",))
     1.8
+elseif any(x -> occursin(x, job_id), ("gpu",))
+    5
 else
     1.1
 end