Torch array context#250

Draft

inducer wants to merge 9 commits into

mainfrom

Commits on Sep 1, 2022

Deprecate with_container_arithmetic's bcast_numpy_array arg

kaushikcfd
authored and
inducer
committed
Implements NumpyArrayContext

kaushikcfd
authored and
inducer
committed
ArrayContainer fixes for numpy arrays as leaf classes

kaushikcfd
authored and
inducer
committed
arithmetic fixes to account for np.ndarray being a leaf array

kaushikcfd
authored and
inducer
committed
test NumpyArrayContext

kaushikcfd
authored and
inducer
committed
test tweaks for NumpyArrayContext

kaushikcfd
authored and
inducer
committed

Commits on Sep 20, 2022

Added new Torch array context
Esteban
committed

Commits on Sep 14, 2023

Merging main: fixing conflicts
Esteban
committed

Commits on Apr 24, 2025

Broadcast torch arrays
ecisneros8
committed