Skip to content

Commit 5353b4d

Browse files
wsmosesavik-pal
authored andcommitted
Update TracedRArray.jl
1 parent d96394c commit 5353b4d

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
@@ -1294,7 +1294,7 @@ function scan_impl!(
12941294
# TODO: fix this for TPUs
12951295
if contains(string(first(Reactant.devices())), "TPU")
12961296
initT = __default_init(T, op)
1297-
if initT != init
1297+
if initT != init && initT != something(init)
12981298
throw(AssertionError("Currently, `init` is not supported on TPUs, provided value $init does not match identity $initT."))
12991299
end
13001300
end

0 commit comments

Comments
 (0)