fix missing method for

FluxML · Jan 21, 2024 · fa7dc1f · fa7dc1f
1 parent 590adf2
commit fa7dc1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/ext/FluxCUDAExt/FluxCUDAExt.jl b/ext/FluxCUDAExt/FluxCUDAExt.jl
@@ -43,6 +43,7 @@ end
 ChainRulesCore.@non_differentiable check_use_cuda()
 
 include("functor.jl")
+include("utils.jl")
 
 function __init__()
     Flux.CUDA_LOADED[] = true

diff --git a/ext/FluxCUDAExt/utils.jl b/ext/FluxCUDAExt/utils.jl
@@ -1 +1 @@
-rng_from_array(::CuArray) = CUDA.default_rng()
+Flux.rng_from_array(::CuArray) = CUDA.default_rng()
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		rng_from_array(::CuArray) = CUDA.default_rng()
		Flux.rng_from_array(::CuArray) = CUDA.default_rng()