mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 11:16:07 +00:00
71ac0bb446
Previously we had a number of APIs in core.py that operated on dimensions and shapes and delegated to instances of DimensionHandler. We remove most of those APIs because by now they ended up doing very little, e.g., `core.sum_dim` was the same as `operator.add`, and `core.sum_shape` was the same as `tuple(map(operator.add))`. We also remove the whole `DimensionHandler` machinery because by now the only other use of non-constant dimensions using this mechanism are the symbolic dimensions used for shape polymorphism, and those support now full operator overloading. (When we introduced `DimensionHandler` we had the masking transformation around that needed it also.)