Fix RNG movement tests

Author: darsnack

test/cuda/layers.jl

@@ -287,6 +287,6 @@ end
     m = layer(0.1; rng = MersenneTwister(123))
     @test_throws ErrorException gpu(m)
     m = layer(0.1; rng = CUDA.default_rng())
-    @test gpu(m).rng === CUDA.default_rng()
+    @test gpu(m).rng isa CUDA.RNG
   end
 end

test/layers/normalisation.jl

@@ -59,7 +59,11 @@ evalwgrad(f, x...) = pullback(f, x...)[1]
 
     # CPU RNGs map onto CPU ok
     if isempty(rng_kwargs)
-      @test cpu(m).rng === Random.default_rng()
+      if VERSION >= v"1.7"
+        @test cpu(m).rng isa Random.TaskLocalRNG
+      else
+        @test cpu(m).rng isa Random._GLOBAL_RNG
+      end
     else
       @test cpu(m).rng === only(values(rng_kwargs))
     end
@@ -101,7 +105,11 @@ end
 
     # CPU RNGs map onto CPU ok
     if isempty(rng_kwargs)
-      @test cpu(m).rng === Random.default_rng()
+      if VERSION >= v"1.7"
+        @test cpu(m).rng isa Random.TaskLocalRNG
+      else
+        @test cpu(m).rng isa Random._GLOBAL_RNG
+      end
     else
       @test cpu(m).rng === only(values(rng_kwargs))
     end