Michael Lazos e72ef4f22a Fix capturable enablement conditions (#125826) ... Only enable capturable if state hasn't been initialized and all parameters are on CUDA. Pull Request resolved: https://github.com/pytorch/pytorch/pull/125826 Approved by: https://github.com/anijain2305 ghstack dependencies: #125825		2024-05-11 06:29:59 +00:00
..
__init__.py	[dynamo] Improve constant-prop for regex/torch.__version__ (#123705)	2024-04-12 19:03:13 +00:00
base.py	[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)	2024-04-17 19:29:34 +00:00
builder.py	[compiled autograd][dynamo] improve lifted autograd.Function.backward handling and fallback to pseudo-eager (#125661)	2024-05-08 21:00:37 +00:00
builtin.py	[dynamo][cpp-guards] Differentiate dict guards wrt to guarding on key order (#124779)	2024-04-25 08:20:35 +00:00
constant.py	[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)	2024-04-17 19:29:34 +00:00
ctx_manager.py	[dynamo] fix crash when context manager is passed to a function (#125321)	2024-05-03 23:01:30 +00:00
dicts.py	[dynamo] Add support for DELETE_SUBSCR (#123526)	2024-04-25 22:07:24 +00:00
distributed.py	support as_python_constant on PlacementClassVariable (#124398)	2024-05-01 21:56:01 +00:00
functions.py	[dynamo] trace through nn parametrize (#125771)	2024-05-09 17:43:48 +00:00
higher_order_ops.py	[dynamo] support torchbind object input (#124978)	2024-05-07 03:02:00 +00:00
iter.py	[dynamo] Improve constant-prop for regex/torch.__version__ (#123705)	2024-04-12 19:03:13 +00:00
lazy.py	[dynamo] fix crash when context manager is passed to a function (#125321)	2024-05-03 23:01:30 +00:00
lists.py	[Dynamo] Fix NamedTuple hasattr bug (#124531)	2024-04-21 04:36:22 +00:00
misc.py	[compiled autograd][dynamo] improve lifted autograd.Function.backward handling and fallback to pseudo-eager (#125661)	2024-05-08 21:00:37 +00:00
nn_module.py	[dynamo] support tracing nn.Module @property that accesses closure cells (#125724)	2024-05-08 23:25:39 +00:00
optimizer.py	Fix capturable enablement conditions (#125826)	2024-05-11 06:29:59 +00:00
script_object.py	[dynamo] support torchbind object input (#124978)	2024-05-07 03:02:00 +00:00
sdpa.py	[dynamo] Refactor reconstruct() not to return anything (#120150)	2024-02-17 17:13:41 +00:00
tensor.py	Add propagate_real_tensors mode for unbacked (#125115)	2024-05-02 15:28:26 +00:00
torch_function.py	[BE]: Try TCH autofixes on torch/ (#125536)	2024-05-05 23:13:59 +00:00
torch.py	[dynamo] support object.__setattr__(obj, name, value) (#124068)	2024-04-17 15:57:14 +00:00
user_defined.py	[dynamo] Use correct source for custom getattr (#125828)	2024-05-09 20:37:23 +00:00