pytorch/torch/_dynamo/variables
2024-07-19 04:42:38 +00:00
..
__init__.py [dynamo][unspecialized-nn-module] Distinguish between user-defined and builtin nn module (#130416) 2024-07-11 14:13:24 +00:00
base.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
builder.py [dynamo] Revert back changes to UnspecializedBuiltinNNModuleVariable (#130991) 2024-07-18 05:01:46 +00:00
builtin.py [dynamo] update call map to allow multiple input parameters (#130748) 2024-07-15 22:16:49 +00:00
constant.py Fix #121334: graph break on constant method call (#130158) 2024-07-12 17:34:46 +00:00
ctx_manager.py Revert "Relax constraints for creating a GenericContextWrappingVariable (#129091)" 2024-07-17 20:59:40 +00:00
dicts.py dynamo add support for defaultdict(set) (#130745) 2024-07-15 22:23:33 +00:00
distributed.py support as_python_constant on PlacementClassVariable (#124398) 2024-05-01 21:56:01 +00:00
functions.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
higher_order_ops.py [Inductor] FlexAttention supports partial masking (#130415) (#130626) 2024-07-14 00:37:26 +00:00
iter.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
lazy.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
lists.py Add python type for list iterators (#130511) 2024-07-12 01:14:18 +00:00
misc.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
nn_module.py [dynamo] Alternative way to skip empty hooks guards on inbuilt nn modules (#131057) 2024-07-19 04:42:38 +00:00
optimizer.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
script_object.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
sdpa.py [3.13, dynamo] codegen PUSH_NULL when callable is codegen'd (#129172) 2024-06-22 17:25:23 +00:00
tensor.py Graph break on tostring for numpy remapping (#131007) 2024-07-18 17:23:41 +00:00
torch_function.py [Dynamo x torch_function] fix incorrect source (#128980) 2024-06-20 14:54:00 +00:00
torch.py [dynamo][dynamic-shapes] Graph break if out shape changes on out= variants (#130074) 2024-07-04 08:36:12 +00:00
user_defined.py Revert "Ensure staticmethods can be allowed in graph (#130882)" 2024-07-18 13:31:30 +00:00