Merge pull request #2065 from Saransh-cpp/non-diff

Make `rng_from_array` non differentiable

Author: ToucheSir

src/utils.jl

@@ -46,6 +46,8 @@ The current defaults are:
 rng_from_array(::AbstractArray) = default_rng_value()
 rng_from_array(::CuArray) = CUDA.default_rng()
 
+@non_differentiable rng_from_array(::Any)
+
 if VERSION >= v"1.7"
   @doc """
       default_rng_value()

test/utils.jl

@@ -799,3 +799,9 @@ end
     end
   end
 end
+
+# make sure rng_from_array is non_differentiable
+@testset "rng_from_array" begin
+  m(x) = (rand(rng_from_array(x)) * x)[1]
+  gradient(m, ones(2))
+end