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.
1 parent bf99c1d commit 82a28a4Copy full SHA for 82a28a4
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