remove LOADED

FluxML · Oct 20, 2024 · 9562811 · 9562811
1 parent 4ad6248
commit 9562811
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/test/ext_amdgpu/basic.jl b/test/ext_amdgpu/basic.jl
@@ -1,5 +1,3 @@
-@test Flux.AMDGPU_LOADED[]
-
 @testset "Basic GPU movement" begin
     @test Flux.gpu(rand(Float64, 16)) isa ROCArray{Float32, 1}
     @test Flux.gpu(rand(Float64, 16, 16)) isa ROCArray{Float32, 2}

diff --git a/test/ext_metal/basic.jl b/test/ext_metal/basic.jl
@@ -1,5 +1,3 @@
-@test Flux.METAL_LOADED[]
-
 @testset "Basic GPU movement" begin
     @test Flux.gpu(rand(Float64, 16)) isa MtlArray{Float32, 1}
     @test Flux.gpu(rand(Float64, 16, 16)) isa MtlArray{Float32, 2}

diff --git a/test/functors.jl b/test/functors.jl
@@ -1,5 +1,5 @@
 x = rand(Float32, 10, 10)
-if !(Flux.CUDA_LOADED[] || Flux.AMDGPU_LOADED[] || Flux.METAL_LOADED[])
+if gpu_device() isa CPUDevice
     @test x === gpu(x)
 end