Elias Ellison 6ffe1da375 Add support for multi device foreach ops (#116064) ... Fix for https://github.com/pytorch/pytorch/issues/102023 Pull Request resolved: https://github.com/pytorch/pytorch/pull/116064 Approved by: https://github.com/mlazos		2023-12-20 04:19:40 +00:00
..
__init__.py	Add Fake Cross Ref Mode, migrate sparse to it (#85382)	2022-09-21 17:15:47 +00:00
fake_tensor.py	Add support for multi device foreach ops (#116064)	2023-12-20 04:19:40 +00:00
fake_utils.py	Expose Flash attn to autograd (#114378)	2023-12-01 23:42:06 +00:00
functional_tensor.py	Support Predispatch functionalization (#113728)	2023-12-19 20:28:35 +00:00
meta_utils.py	Expand dynamic dims support for traceable subclasses (#114311)	2023-12-05 21:09:25 +00:00
schema_check_mode.py	Use pytree.tree_leaves everywhere (#112324)	2023-10-30 03:39:04 +00:00