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

Simplify Numba conversion of Composite Op

上级 2ae132e4
......@@ -602,13 +602,11 @@ def numba_funcify_CAReduce(op, node, **kwargs):
@numba_funcify.register(Composite)
def numba_funcify_Composite(op, node, **kwargs):
numba_impl = numba.njit(numba_funcify(op.fgraph, **kwargs))
@numba.njit
def composite(*args):
return numba_impl(*args)[0]
return composite
signature = create_numba_signature(node, force_scalar=True)
composite_fn = numba.njit(signature)(
numba_funcify(op.fgraph, squeeze_output=True, **kwargs)
)
return composite_fn
def create_index_func(node, objmode=False):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论