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