提交 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}): ...@@ -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) 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) @numba_funcify.register(Switch)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论