Skip to content

Commit 5d1ecd3

Browse files
committed
more tpu fixes
1 parent a536bd1 commit 5d1ecd3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/basic.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ end
420420

421421
@testset "Complex runtime: $CT" for CT in (ComplexF32, ComplexF64)
422422
# complex f64 not supported on tpu
423-
if CT == ComplexF32 || !contains(string(Reactant.devices()[1]), "tpu")
423+
if CT == ComplexF32 || !contains(string(Reactant.devices()[1]), "TPU")
424424
a = Reactant.to_rarray(ones(CT, 2))
425425
b = Reactant.to_rarray(ones(CT, 2))
426426
c = Reactant.compile(+, (a, b))(a, b)
@@ -784,7 +784,7 @@ end
784784
x = Reactant.to_rarray([1.0, NaN, Inf, -Inf, NaN])
785785
@test @jit(isfinite.(x)) == [true, false, false, false, false]
786786

787-
if !contains(string(Reactant.devices()[1]), "tpu")
787+
if !contains(string(Reactant.devices()[1]), "TPU")
788788
x = Reactant.to_rarray([1.0, NaN, Inf, -Inf, NaN] .* im)
789789
@test @jit(isfinite.(x)) == [true, false, false, false, false]
790790
end

0 commit comments

Comments
 (0)