Skip to content

Commit d504d16

Browse files
committed
Only free if the context isn't destroyed.
1 parent 269a2dd commit d504d16

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/compiler/execution.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,10 +204,9 @@ mutable struct HostKernel{F,TT} <: AbstractKernel{F,TT}
204204
kernel = new{F,TT}(ctx, mod, fun, random_state)
205205
finalizer(kernel) do k
206206
if k.random_state isa Mem.DeviceBuffer
207-
Mem.free(k.random_state)
207+
@context! skip_destroyed=true k.ctx Mem.free(k.random_state; stream_ordered=false)
208208
end
209209
end
210-
return kernel
211210
end
212211
end
213212

0 commit comments

Comments
 (0)