提交 175e3e86 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Specify signatures during Numba conversion of scalar Ops

上级 f7a3dd6f
......@@ -371,7 +371,9 @@ def {scalar_op_fn_name}({input_names}):
"""
scalar_op_fn = compile_function_src(scalar_op_src, scalar_op_fn_name, global_env)
return numba.njit(scalar_op_fn)
signature = create_numba_signature(node, force_scalar=True)
return numba.njit(signature)(scalar_op_fn)
@numba_funcify.register(Switch)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论