Numba: fuse AdvancedSubtensor->Elemwise->AdvancedIncSubtensor#2015

Draft

ricardoV94 wants to merge 7 commits into

pymc-devs:mainfrom

ricardoV94:gather_scatter_fusion

Commits on Apr 5, 2026

Warn when copy(delete_updates=True) leaves inplace-destroyed shared variables
ricardoV94
committed
Elemwise: fuse sibling subgraphs in FusionOptimizer
ricardoV94
committed
Numba: fuse AdvancedSubtensor1 indexed reads into Elemwise
ricardoV94
committed
Numba: fuse AdvancedIncSubtensor1 indexed updates into Elemwise
ricardoV94
committed
Numba: generalize IndexedElemwise to arbitrary-axis and multi-index
ricardoV94
committed
Numba: fuse ND and 0-d integer indices into IndexedElemwise
ricardoV94
committed
Numba: allow broadcast scatter in IndexedElemwise
ricardoV94
committed