3 Commits

Author	SHA1	Message	Date
ydwu4	c71828b097	Lift non-FakeTensor restriction for compile (#107042) ... Currently, we have the assertion that dynamo won't accept FakeTensor input unless we're exporting. This PR try to remove this restriction to finish https://github.com/pytorch/pytorch/pull/105679. Pull Request resolved: https://github.com/pytorch/pytorch/pull/107042 Approved by: https://github.com/ezyang, https://github.com/zou3519	2023-08-15 20:58:56 +00:00
Michael Lazos	05eea20eb9	[dynamo] Simulate torch function enablement state (#105091) ... Part of https://github.com/pytorch/pytorch/issues/93723 Pull Request resolved: https://github.com/pytorch/pytorch/pull/105091 Approved by: https://github.com/voznesenskym, https://github.com/anijain2305	2023-07-13 17:42:20 +00:00
Michael Lazos	dfe7a1e089	[dynamo] Support wrapping + returning tensor subclasses (#104802) ... as title - used for [tracing the FSDP collectives](d8cb80e382/torch/distributed/_functional_collectives.py (L425)) Pull Request resolved: https://github.com/pytorch/pytorch/pull/104802 Approved by: https://github.com/jansel	2023-07-09 22:16:10 +00:00