pytorch/torch/_dynamo/variables
2024-11-05 19:11:41 +00:00
..
__init__.py Add host-side Triton TMA support to Dynamo (#137677) 2024-10-16 02:18:48 +00:00
base.py [dynamo][NFC] Move MutationType classes into variables/base.py (#139340) 2024-11-05 19:11:41 +00:00
builder.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
builtin.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
constant.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
ctx_manager.py [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114) 2024-10-09 02:29:40 +00:00
dicts.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
distributed.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
functions.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
higher_order_ops.py [hop free symbols][refactor] make create_graph_input always take example_value (#138428) 2024-11-04 22:47:49 +00:00
iter.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
lazy.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
lists.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
misc.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
nn_module.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
optimizer.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
script_object.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
sdpa.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
tensor.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
torch_function.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
torch.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
user_defined.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00