pytorch/torch/_dynamo/variables
2024-07-11 14:13:24 +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 Add VariableTracker.debug_repr (#126299) 2024-05-15 23:55:29 +00:00
builder.py [dynamo][tuple] Optimize guard for small tuples - helps conv2d guards (#130400) 2024-07-11 14:13:24 +00:00
builtin.py [easy][dynamo] Remove try except from call_getattr (#129217) 2024-06-21 23:56:00 +00:00
constant.py [dynamo] Support enum construction (#127364) 2024-05-29 08:09:21 +00:00
ctx_manager.py [3.13, dynamo] codegen PUSH_NULL when callable is codegen'd (#129172) 2024-06-22 17:25:23 +00:00
dicts.py [dynamo][compile-time][inlining-inbuilt-nn-modules] Manually implement nn.Module._call_impl (#129285) 2024-06-25 01:31:26 +00:00
distributed.py support as_python_constant on PlacementClassVariable (#124398) 2024-05-01 21:56:01 +00:00
functions.py Refactor TritonKernelVariable's logic so it can be shared (#130177) 2024-07-10 03:09:29 +00:00
higher_order_ops.py Revert "[cond] inlining into one of the branches when pred is a python constant (#128709)" 2024-07-10 17:47:19 +00:00
iter.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
lazy.py [dynamo] fix crash when context manager is passed to a function (#125321) 2024-05-03 23:01:30 +00:00
lists.py Bug fixes for disabling 0/1 specialization on plain int (#129961) 2024-07-02 23:19:48 +00:00
misc.py [dynamo][compile-time][inlining-inbuilt-nn-modules] Manually implement nn.Module._call_impl (#129285) 2024-06-25 01:31:26 +00:00
nn_module.py [dynamo][unspecialized-nn-module] Distinguish between user-defined and builtin nn module (#130416) 2024-07-11 14:13:24 +00:00
optimizer.py Fix capturable enablement conditions (#125826) 2024-05-11 06:29:59 +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 [HOP] Use user directed names for variables where possible (#130271) 2024-07-10 13:59:20 +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 [dynamo][user-defined] Simplify and improve scope of UserDefinedObject var_getattr (#130169) 2024-07-08 04:10:56 +00:00