pytorch/torch/fx/experimental
Klint Qinami 2d98a1caf5 [MTIA] Map names to operand indices when folding submodules (#150692)
When replacing placeholders with getattrs during constant folding, we can have an argument and parameter name mismatch. In fact, there is no guarantee that the parameter name is equivalent to the argument name used in the module call.

Differential Revision: D72415970

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150692
Approved by: https://github.com/jfix71
2025-04-06 03:11:14 +00:00
..
migrate_gradual_types PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
shape_inference PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
unification Remove code for Python < 3.9 (#147097) 2025-02-14 03:22:49 +00:00
__init__.py
_backward_state.py
_config.py [dynamic shapes] add backed_size_oblivious option (#148696) 2025-03-11 21:52:34 +00:00
_constant_symnode.py Switch to using Python nested int (#141166) 2024-12-02 19:17:30 +00:00
_dynamism.py Make dynamism code robust to NotImplementedException (#148823) 2025-03-14 23:38:19 +00:00
accelerator_partitioner.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
const_fold.py [MTIA] Map names to operand indices when folding submodules (#150692) 2025-04-06 03:11:14 +00:00
debug.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
graph_gradual_typechecker.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
merge_matmul.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
meta_tracer.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
normalize.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
optimization.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
partitioner_utils.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
proxy_tensor.py Rename node.meta["arg_kwarg_vals"] to node.meta["eager_input_vals"] (#148092) 2025-04-02 13:18:04 +00:00
recording.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
refinement_types.py remove allow-untyped-defs from fx/experimental/refinement_types.py (#143868) 2024-12-27 01:00:45 +00:00
rewriter.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
schema_type_annotation.py PEP585 update - torch/fx (#145166) 2025-01-20 18:11:54 +00:00
sym_node.py [dynamic shapes] C++ bindings for guard_or_false/true (#150148) 2025-03-31 17:04:25 +00:00
symbolic_shapes.py [dynamic shapes] allow duck typing for 0/1 (#150222) 2025-04-04 03:24:46 +00:00
unify_refinements.py
validator.py [BE] Fix edge case in translation validation bisector (#145414) 2025-01-23 17:35:28 +00:00