Skip to content

Commit d42ba71

Browse files
committed
more tpu fixes
1 parent 11d0acc commit d42ba71

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/autodiff.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,7 @@ end
195195
@test res2 4 * 3 * 3.1^2
196196
end
197197

198+
if !contains(string(Reactant.devices()[1]), "TPU")
198199
@testset "Seed initialization of Complex arrays on matmul: Issue #593" begin
199200
a = ones(ComplexF64, 2, 2)
200201
b = 2.0 * ones(ComplexF64, 2, 2)
@@ -208,6 +209,7 @@ end
208209
(res.derivs[2] 2ones(2, 2))
209210
end
210211
end
212+
end
211213

212214
@testset "onehot" begin
213215
x = Reactant.to_rarray(rand(3, 4))

0 commit comments

Comments
 (0)