fix

FluxML · Oct 11, 2024 · 61312a8 · 61312a8
1 parent e28ca8f
commit 61312a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/functor.jl b/src/functor.jl
@@ -201,7 +201,7 @@ end
 const GPU_BACKENDS = ("CUDA", "AMDGPU", "Metal", "CPU")
 # const GPU_BACKEND = load_preference(MLDataDevices, "gpu_backend", "CUDA")
 # https://github.com/JuliaPackaging/Preferences.jl/issues/39
-const GPU_BACKEND = @load_preference!("Flux", "gpu_backend", "CUDA")
+const GPU_BACKEND = @load_preference("gpu_backend")
 
 """
     gpu(m)