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()