Skip to content

Commit b6f4c8c

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_ModelApplicator.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ end;
3333
@test p r
3434
gdev = gpu_device()
3535
#cdev = cpu_device()
36-
if MLDataDevices.functional(gdev)
36+
if gdev isa MLDataDevices.AbstractGPUDevice
3737
g_gpu = g |> gdev
3838
@test g_gpu.μ isa GPUArraysCore.AbstractGPUArray
3939
r_gpu = r |> gdev

0 commit comments

Comments
 (0)