Michael Lazos 7c23fed12c Move step to cpu if state is already initialized (#123618) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/123618 Approved by: https://github.com/anijain2305 ghstack dependencies: #123496, #123497, #123551, #123552		2024-04-09 09:04:18 +00:00
..
__init__.py	[dynamo] Support warnings.catch_warnings (#123511)	2024-04-08 22:27:46 +00:00
base.py	[dynamo] Support __next__ on UserDefinedObjectVariable (#122565)	2024-03-31 19:00:03 +00:00
builder.py	Revert "Ignore logging.Logger.* calls during dynamo export (#123402)"	2024-04-09 06:28:12 +00:00
builtin.py	[dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318)	2024-04-07 21:06:52 +00:00
constant.py	[dynamo] Remove VariableTracker.parents_tracker (#122058)	2024-03-19 04:23:24 +00:00
ctx_manager.py	[dynamo] Support warnings.catch_warnings (#123511)	2024-04-08 22:27:46 +00:00
dicts.py	[dynamo, 3.12] remove LOAD_METHOD, update LOAD_ATTR (#122356)	2024-03-27 20:39:39 +00:00
distributed.py	Don't create world pg variable out of thin air when rewriting c10d collectives (#122561)	2024-03-26 20:12:08 +00:00
functions.py	[dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318)	2024-04-07 21:06:52 +00:00
higher_order_ops.py	Use graph.find_nodes in dynamo (#122257)	2024-04-07 18:51:18 +00:00
iter.py	[dynamo] Support __next__ on UserDefinedObjectVariable (#122565)	2024-03-31 19:00:03 +00:00
lazy.py	[dynamo] Replace VariableTracker.apply with visit/realize_all (#122218)	2024-03-20 07:53:18 +00:00
lists.py	[dynamo] Support __next__ on UserDefinedObjectVariable (#122565)	2024-03-31 19:00:03 +00:00
misc.py	Revert "Ignore logging.Logger.* calls during dynamo export (#123402)"	2024-04-09 06:28:12 +00:00
nn_module.py	[dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318)	2024-04-07 21:06:52 +00:00
optimizer.py	Move step to cpu if state is already initialized (#123618)	2024-04-09 09:04:18 +00:00
sdpa.py	[dynamo] Refactor reconstruct() not to return anything (#120150)	2024-02-17 17:13:41 +00:00
tensor.py	dynamo: support placement kwargs for DTensor.to_local() (#119947)	2024-03-22 14:42:27 +00:00
torch_function.py	[dynamo] Optimize SourcelessBuilder (#122063)	2024-03-19 04:23:30 +00:00
torch.py	Teach dynamo about torch.func.jvp (#119926)	2024-03-22 20:25:47 +00:00
user_defined.py	[dynamo] Support warnings.catch_warnings (#123511)	2024-04-08 22:27:46 +00:00