fix

FluxML · Jul 19, 2023 · c274416 · c274416
1 parent 08957df
commit c274416
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/test/functors.jl b/test/functors.jl
@@ -1,10 +1,4 @@
 x = rand(Float32, 10, 10)
-if Flux.CUDA_LOADED[]
-    @test x !== gpu(x)
-elseif Flux.AMD_LOADED[]
-    @test x !== gpu(x)
-elseif Flux.METAL_LOADED[]
-    @test x !== gpu(x)
-else
+if !(Flux.CUDA_LOADED[] || Flux.AMD_LOADED[] || Flux.METAL_LOADED[])
     @test x === gpu(x)
 end