relax identity test for devices (#2544)

FluxML · Dec 7, 2024 · 5cc50e9 · 5cc50e9
1 parent 4edc2ac
commit 5cc50e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/functors.jl b/test/functors.jl
@@ -1,6 +1,6 @@
 x = rand(Float32, 10, 10)
 if gpu_device() isa CPUDevice
-    @test x === gpu(x)
+    @test x == gpu(x)
 end
 
 dev = Flux.cpu_device()