Skip to content

Commit 48ca5eb

Browse files
wsmosesavik-pal
authored andcommitted
Update TracedRArray.jl
1 parent 0fb1576 commit 48ca5eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/TracedRArray.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1295,7 +1295,7 @@ function scan_impl!(
12951295
# TODO: fix this for TPUs
12961296
if contains(string(first(Reactant.devices())), "TPU")
12971297
initT = __default_init(T, op)
1298-
if initT != init
1298+
if initT != init && initT != something(init)
12991299
throw(AssertionError("Currently, `init` is not supported on TPUs, provided value $init does not match identity $initT."))
13001300
end
13011301
end

0 commit comments

Comments
 (0)