Skip to content

Commit

Permalink
Do not reuse outer-graph storage in Numba conversions of Composites
Browse files Browse the repository at this point in the history
  • Loading branch information
brandonwillard committed Nov 10, 2022
1 parent d10eafa commit 4a6ecdf
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions aesara/link/numba/dispatch/scalar.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,6 +221,7 @@ def clip(_x, _min, _max):
@numba_funcify.register(Composite)
def numba_funcify_Composite(op, node, **kwargs):
signature = create_numba_signature(node, force_scalar=True)
_ = kwargs.pop("storage_map", None)
composite_fn = numba_basic.numba_njit(signature, fastmath=config.numba__fastmath)(
numba_funcify(op.fgraph, squeeze_output=True, **kwargs)
)
Expand Down

0 comments on commit 4a6ecdf

Please sign in to comment.