Update Race.jl

CliMA · Jul 25, 2024 · d0cbc6f · d0cbc6f
1 parent 1ae317a
commit d0cbc6f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/TestKernels/Race.jl b/TestKernels/Race.jl
@@ -10,8 +10,10 @@ function index_fun(arr; backend=get_backend(arr))
 	fill!(out, 0)
 	kernel! = my_kernel!(backend)
 	kernel!(out, arr, ndrange=(size(arr, 1), size(arr, 2)))
+        KernelAbstractions.synchronize(backend)
         @time for i = 1 : 200
 	    kernel!(out, arr, ndrange=(size(arr, 1), size(arr, 2)))
+            KernelAbstractions.synchronize(backend)
         end    
 	return out
 end