pytorch/torch/_dynamo/variables
2024-09-14 18:52:22 +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 [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) 2024-09-14 18:52:22 +00:00
builtin.py [dynamo] reland map/zip iterator related changes (#135074) 2024-09-06 20:38:02 +00:00
constant.py Revert "[dynamo] Fix support for classmethod(property(...)) (#134968)" 2024-09-13 12:29:03 +00:00
ctx_manager.py [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) 2024-09-14 18:52:22 +00:00
dicts.py [dynamo][dicts][nv-embed] Support update with kwargs (#135588) 2024-09-10 23:50:23 +00:00
distributed.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
functions.py [dynamo] Bug fix for _torchdynamo_inline source handling (#135612) 2024-09-12 04:05:08 +00:00
higher_order_ops.py Implementation of scan (#134102) 2024-09-10 04:51:16 +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] Trace enter/exit of TorchFunctionModes (#135422) 2024-09-14 18:52:22 +00:00