pytorch/torch/_dynamo/variables
2025-01-13 22:24:56 +00:00
..
__init__.py [dynamo][dicts] Guarding lazily on dict keys (#143997) 2025-01-08 03:56:33 +00:00
base.py dynamo tracing perf: slight improvement on __instancecheck__: 47.77 -> 47.62 (#143064) 2024-12-19 09:19:35 +00:00
builder.py [dynamo][dicts] Guarding lazily on dict keys (#143997) 2025-01-08 03:56:33 +00:00
builtin.py [dynamo][dicts] Consolidate dict(..) construction (#144342) 2025-01-13 22:24:56 +00:00
constant.py [dynamo] Expand support of enum attribute access (#142268) 2024-12-10 19:32:40 +00:00
ctx_manager.py [FSDP2] Move to public torch.distributed.fsdp (#141868) 2024-12-07 01:24:28 +00:00
dicts.py [dynamo][dicts] Guarding lazily on dict keys (#143997) 2025-01-08 03:56:33 +00:00
distributed.py [dynamo] Fix VariableBuilder._wrap on frozenset and enforce invariants on ConstantVariable (#141504) 2024-11-27 21:58:35 +00:00
functions.py [user triton] add support for @triton.heuristics after @triton.autotune (#142208) 2025-01-11 02:18:26 +00:00
higher_order_ops.py [Dynamo] Supports autograd.Function forward returns constant (#144597) 2025-01-12 03:53:10 +00:00
iter.py [dynamo][BE] move zip_longest polyfill to submodule polyfills.itertools (#144067) 2025-01-03 08:08:31 +00:00
lazy.py Migrate from Tuple -> tuple in torch/_dynamo (#144261) 2025-01-10 07:45:57 +00:00
lists.py [dynamo] Use proper item source for NamedTupleVariable (#142437) 2024-12-10 19:23:48 +00:00
misc.py [dynamo] Avoid graph break on updates to obj.__dict__ (#144419) 2025-01-13 21:04:10 +00:00
nn_module.py [dynamo][dicts] Guarding lazily on dict keys (#143997) 2025-01-08 03:56:33 +00:00
optimizer.py [dynamo] Separate out GetItemSource and DictGetItemSource (#143926) 2025-01-01 02:39:41 +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 Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
torch_function.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
torch.py [export] add is_exporting flag (#142425) 2024-12-18 21:36:28 +00:00
user_defined.py [dynamo][dicts] Consolidate dict(..) construction (#144342) 2025-01-13 22:24:56 +00:00