From 2b621bccdedcc62aee8babf56b6c3509ed32cff6 Mon Sep 17 00:00:00 2001 From: "Brandon T. Willard" Date: Thu, 10 Nov 2022 14:18:15 -0600 Subject: [PATCH] Do not reuse outer-graph storage in Numba conversions of Composites --- aesara/link/numba/dispatch/scalar.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aesara/link/numba/dispatch/scalar.py b/aesara/link/numba/dispatch/scalar.py index 28031ea988..f9dcb08a36 100644 --- a/aesara/link/numba/dispatch/scalar.py +++ b/aesara/link/numba/dispatch/scalar.py @@ -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) )