Michael Lazos f302a0d380 Re-enable SGD (#117434) ... Re-enables the SGD optimizer now that compile times are more reasonable. [Benchmark run](https://github.com/pytorch/pytorch/actions/runs/7511073761) Pull Request resolved: https://github.com/pytorch/pytorch/pull/117434 Approved by: https://github.com/anijain2305, https://github.com/janeyx99		2024-01-19 04:28:50 +00:00
..
__init__.py	Implement set in terms of dict (#110524)	2024-01-18 09:36:41 +00:00
base.py	[dynamo] add common methods to DistributedVariable (#117590)	2024-01-18 17:32:31 +00:00
builder.py	[dynamo] Extend LazyVariableTracker to tuples (#117426)	2024-01-18 15:51:28 +00:00
builtin.py	[dynamo] make ConstantSource propagate through built-in ops for TensorVariable (#117704)	2024-01-18 20:18:34 +00:00
constant.py	[dynamo] Improve support for dynamic shapes str.format and _assert (#115203)	2023-12-06 04:54:45 +00:00
ctx_manager.py	Reconstruct streams via global registration, temporary impl to unblock FSDP (#117386)	2024-01-13 07:03:33 +00:00
dicts.py	[dynamo] Add DictView variable tracker (#108420)	2024-01-18 09:37:33 +00:00
distributed.py	[dynamo] add common methods to DistributedVariable (#117590)	2024-01-18 17:32:31 +00:00
functions.py	Impl. call_hasattr for BaseUserFunctionVariable (#116049)	2024-01-09 22:58:58 +00:00
higher_order_ops.py	[HigherOrderOp][BE] factor out merge_graph_inputs (#116912)	2024-01-19 00:35:26 +00:00
iter.py	[BE]: Enable F821 and fix bugs (#116579)	2024-01-01 08:40:46 +00:00
lazy.py	[dynamo] LazyVariable - redirect __str__ to the realized variable __str__ (#117583)	2024-01-17 21:12:12 +00:00
lists.py	[dynamo] Add hasattr support for TupleVariable (#117694)	2024-01-18 07:47:43 +00:00
misc.py	[Dynamo] Trace autograd.function in dynamo when inputs require grad (#116358) (#116897)	2024-01-16 03:57:13 +00:00
nn_module.py	Support NNModules as dict keys (#116723)	2024-01-09 03:32:47 +00:00
optimizer.py	Re-enable SGD (#117434)	2024-01-19 04:28:50 +00:00
tensor.py	[easy]Remove specialized value (#112252)	2024-01-18 09:34:50 +00:00
torch_function.py	Fix torch function kwarg dispatch (#117083)	2024-01-10 10:55:10 +00:00
torch.py	[Dynamo] Make profiler function will be ignored warn only once (#117585)	2024-01-17 04:05:45 +00:00
user_defined.py	Reland #111196 (take 4) "Support tensors as Dict keys" (#116934)	2024-01-07 01:37:26 +00:00