bobrenjc93 4708cfdbd9 Support whitelist of dynamic sources (#147979) ...

This PR introduces the ability to whitelist sources as dynamic. This is particularly useful for large models with graph breaks, as you can keep the dynamism across graph breaks since source names stay consistent. Additionally you can use this to mark ints as dynamic.

NB: I intentionally didn't complicate the interface by supporting specification of per dimension dynamism. There is virtue in keeping true to the standard way of representing sources (eg. L['x']). If we find in practice that we need more more fine grained control, we can explore further affordances at that time.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147979
Approved by: https://github.com/Mingming-Ding

2025-02-28 15:43:14 +00:00

..

__init__.py

PEP585: More UP006 fixes (#146392)

2025-02-20 06:18:13 +00:00

_cache.py

PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)

2025-01-20 22:37:26 +00:00

config.py

Support whitelist of dynamic sources (#147979)

2025-02-28 15:43:14 +00:00