File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ module DynamicExpressionsBumperExt
22
33using Bumper: @no_escape , @alloc
44using DynamicExpressions: OperatorEnum, AbstractExpressionNode, tree_mapreduce
5- using DynamicExpressions. UtilsModule: ResultOk, counttuple
5+ using DynamicExpressions. UtilsModule: ResultOk, counttuple, is_bad_array
66
77import DynamicExpressions. ExtensionInterfaceModule: bumper_eval_tree_array
88
@@ -35,7 +35,7 @@ function bumper_eval_tree_array(
3535 if M == 1
3636 if cumulators[1 ]. ok
3737 out = dispatch_kern1! (operators. unaops, branch. op, cumulators[1 ]. x)
38- ResultOk (out, isfinite ( sum (yi -> yi * zero (yi), out) ))
38+ ResultOk (out, ! is_bad_array ( out))
3939 else
4040 cumulators[1 ]
4141 end
@@ -47,7 +47,7 @@ function bumper_eval_tree_array(
4747 cumulators[1 ]. x,
4848 cumulators[2 ]. x,
4949 )
50- ResultOk (out, isfinite ( sum (yi -> yi * zero (yi), out) ))
50+ ResultOk (out, ! is_bad_array ( out))
5151 elseif cumulators[1 ]. ok
5252 cumulators[2 ]
5353 else
You can’t perform that action at this time.
0 commit comments