Merge 48a8b4e into 849beb7

SciML · Jul 28, 2020 · 394ac19 · 394ac19
2 parents 849beb7 + 48a8b4e
commit 394ac19
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/DiffEqGPU.jl b/src/DiffEqGPU.jl
@@ -32,8 +32,9 @@ end
 end
 
 @kernel function continuous_affect!_kernel(affect!,event_idx,u,t,p)
-    i = @index(Global, Linear)
-    @views @inbounds affect!(FakeIntegrator(u[:,i],t,p[:,i]))
+    for i in event_idx
+        @views @inbounds affect!(FakeIntegrator(u[:,i],t,p[:,i]))
+    end
 end
 
 maxthreads(::CPU) = 1024