Skip to content

Commit a1e78f7

Browse files
committed
alter check on GPU device from function to testing type another test
1 parent b6f4c8c commit a1e78f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_HybridProblem.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ import Flux
149149
@test gr[1] isa Vector
150150

151151
gdev = gpu_device()
152-
if MLDataDevices.functional(gdev)
152+
if gdev isa MLDataDevices.AbstractGPUDevice
153153
@testset "neg_elbo_transnorm_gf gpu" begin
154154
g, ϕg0 = begin
155155
n_covar = size(xM, 1)

0 commit comments

Comments
 (0)