Only free if the context isn't destroyed.

JuliaGPU · Mar 19, 2021 · 30b661f · 30b661f
1 parent 269a2dd
commit 30b661f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compiler/execution.jl b/src/compiler/execution.jl
@@ -204,7 +204,7 @@ mutable struct HostKernel{F,TT} <: AbstractKernel{F,TT}
         kernel = new{F,TT}(ctx, mod, fun, random_state)
         finalizer(kernel) do k
             if k.random_state isa Mem.DeviceBuffer
-                Mem.free(k.random_state)
+                @context! skip_destroyed=true Mem.free(k.random_state)
             end
         end
         return kernel