From 5cc50e923df5ef5ca90344923b5dceb8498a4c5a Mon Sep 17 00:00:00 2001
From: Carlo Lucibello <carlo.lucibello@unibocconi.it>
Date: Sat, 7 Dec 2024 17:36:07 +0100
Subject: [PATCH] relax identity test for devices (#2544)

---
 test/functors.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/functors.jl b/test/functors.jl
index 111da50ea8..d1a8427931 100644
--- 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()