pytorch/torch/_dynamo/variables
2024-11-01 21:02:32 +00:00
..
__init__.py Add host-side Triton TMA support to Dynamo (#137677) 2024-10-16 02:18:48 +00:00
base.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
builder.py [Dynamo] Refactor wrap_fx_proxy (#138933) 2024-10-28 08:05:33 +00:00
builtin.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
constant.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
ctx_manager.py
dicts.py Avoid calling realize() on LazyVariableTracker on reconstruct (#138495) 2024-10-25 19:01:15 +00:00
distributed.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
functions.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
higher_order_ops.py [invoke_subgraph] Generate fake_inputs correctly (#139130) 2024-11-01 21:02:32 +00:00
iter.py [BE]: Add better optional typing (#138426) 2024-10-27 14:19:00 +00:00
lazy.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
lists.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
misc.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
nn_module.py [dynamo][guards] Skip guards on empty nn module hooks (#138942) 2024-10-29 02:11:47 +00:00
optimizer.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
script_object.py
sdpa.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
tensor.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
torch.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
torch_function.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
user_defined.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00