You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
We could adopt the approach that the C-backend takes, which is to just use pointers to the storage list created by PyTensor function, for these constants. We only inline scalars in the C-backend codegen. This would work-around all the issues listed above.
On the downside, I'm not sure how hard it would make things for nutpie. CC @aseyboldt
Description
In the numba backend (and all Jitted backends), we massage constants into global variables of the codegen function.
This poses has a couple of issues for the Numba backend:
We could adopt the approach that the C-backend takes, which is to just use pointers to the storage list created by PyTensor function, for these constants. We only inline scalars in the C-backend codegen. This would work-around all the issues listed above.
On the downside, I'm not sure how hard it would make things for nutpie. CC @aseyboldt