We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents bf99c1d + 82a28a4 commit 7fac139Copy full SHA for 7fac139
src/backends/cuda.jl
@@ -6,11 +6,9 @@ const FREE_STREAMS = CuStream[]
6
const STREAMS = CuStream[]
7
const STREAM_GC_THRESHOLD = Ref{Int}(16)
8
9
-@init begin
10
- if haskey(ENV, "KERNELABSTRACTIONS_STREAMS_GC_THRESHOLD")
11
- global STREAM_GC_THRESHOLD[] = parse(Int, ENV["KERNELABSTRACTIONS_STREAMS_GC_THRESHOLD"])
12
- end
13
-
+# This code is loaded after an `@init` step
+if haskey(ENV, "KERNELABSTRACTIONS_STREAMS_GC_THRESHOLD")
+ global STREAM_GC_THRESHOLD[] = parse(Int, ENV["KERNELABSTRACTIONS_STREAMS_GC_THRESHOLD"])
14
end
15
16
## Stream GC
0 commit comments