提交 2ae132e4 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove unnecessary np.all from Numba Switch conversion

上级 78c94889
......@@ -378,7 +378,10 @@ def {scalar_op_fn_name}({input_names}):
def numba_funcify_Switch(op, node, **kwargs):
@numba.njit
def switch(condition, x, y):
return x if np.all(condition) else y
if condition:
return x
else:
return y
return switch
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论