reintroduce GPU_BACKEND

FluxML · Nov 1, 2024 · 1763efe · 1763efe
1 parent eb9ec4a
commit 1763efe
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/deprecations.jl b/src/deprecations.jl
@@ -145,6 +145,12 @@ function supported_devices()
   return MLDataDevices.supported_gpu_backends()
 end
 
+# This was previosly documented. 
+# As of v0.14.23 we silently deprecate it. 
+# Later we will deprecate it loudly and then remove it.
+const GPU_BACKEND = @load_preference("gpu_backend", "CUDA") 
+
+
 # v0.15 deprecations
 
 # Enable these when 0.15 is released, and delete const ClipGrad = Optimise.ClipValue etc: