pytorch/torch/_dynamo/variables
2024-09-18 04:47:51 +00:00
..
__init__.py [dynamo] support torch.nn.attention.sdpa_kernel context manager (#135404) 2024-09-12 22:04:48 +00:00
base.py [dynamo] reland map/zip iterator related changes (#135074) 2024-09-06 20:38:02 +00:00
builder.py Revert "Optimize dict reconstruct to not codegen untouched values (#134876)" 2024-09-17 13:00:01 +00:00
builtin.py [dynamo] simplify implementation for builtins.sum (#133779) 2024-09-16 04:53:06 +00:00
constant.py [dynamo] Fix support for classmethod(property(...)) (#134968) 2024-09-18 04:47:51 +00:00
ctx_manager.py [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) 2024-09-14 18:52:22 +00:00
dicts.py Revert "Optimize dict reconstruct to not codegen untouched values (#134876)" 2024-09-17 13:00:01 +00:00
distributed.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
functions.py [Traceable FSDP2] Ignore FSDP2 forward hook side-effects in AC; Support FSDP2 + AC (#134997) 2024-09-15 02:00:17 +00:00
higher_order_ops.py [Traceable FSDP2] Ignore FSDP2 forward hook side-effects in AC; Support FSDP2 + AC (#134997) 2024-09-15 02:00:17 +00:00
iter.py [dynamo] reland map/zip iterator related changes (#135074) 2024-09-06 20:38:02 +00:00
lazy.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
lists.py [dynamo] reland map/zip iterator related changes (#135074) 2024-09-06 20:38:02 +00:00
misc.py Implement VariableTracker.python_type() (#134215) 2024-09-05 16:35:47 +00:00
nn_module.py [fbode-testing][dynamo][reland][inline-inbuilt-nn-modules] Mark attri… (#134136) 2024-08-22 17:54:58 +00:00
optimizer.py [Inductor UT] Reuse inductor UT for intel GPU test/inductor/test_compiled_optimizers.py (#133083) 2024-08-17 01:15:26 +00:00
script_object.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
sdpa.py Grouped Query Attention (#132689) 2024-08-07 05:35:36 +00:00
tensor.py [Traceable FSDP2][Partitioner] Must save AC output if output has a backward hook (#135727) 2024-09-14 08:45:58 +00:00
torch_function.py [Dynamo] Remove ignored modes workaround (#135502) 2024-09-14 18:52:22 +00:00
torch.py [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) 2024-09-14 18:52:22 +00:00
user_defined.py [dynamo] Fix support for classmethod(property(...)) (#134968) 2024-09-18 04:47:51 +00:00