Thomas Bohnstingl 64dc30c213 [HOP, map] Rework of map autograd to the new interface (#153343) ... This PR reworks the current autograd implementation of map to the new interface. @pytorchbot label "topic: not user facing" Pull Request resolved: https://github.com/pytorch/pytorch/pull/153343 Approved by: https://github.com/ydwu4		2025-08-06 23:02:42 +00:00
..
__init__.py	[export][cond] support merging constant ints as unbacked symint (#152742)	2025-05-22 17:25:38 +00:00
_invoke_quant.py	[BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730)	2025-02-20 02:30:36 +00:00
aoti_call_delegate.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
associative_scan.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
auto_functionalize.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
base_hop.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
cond.py	[HOP, map] Rework of map autograd to the new interface (#153343)	2025-08-06 23:02:42 +00:00
effects.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
executorch_call_delegate.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
flat_apply.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
flex_attention.py	Make Q Indices optional (#157997)	2025-07-12 00:16:20 +00:00
foreach_map.py	[BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730)	2025-02-20 02:30:36 +00:00
hints_wrap.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
invoke_subgraph.py	[Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491)	2025-07-30 22:57:50 +00:00
map.py	[HOP, map] Rework of map autograd to the new interface (#153343)	2025-08-06 23:02:42 +00:00
out_dtype.py	Enable _int_mm on Intel GPU (#157769)	2025-08-02 05:16:01 +00:00
run_const_graph.py	remove allow-untyped-defs from torch/_higher_order_ops/run_const_graph.py (#157847)	2025-07-12 15:42:12 +00:00
scan.py	[HOP, map] Rework of map autograd to the new interface (#153343)	2025-08-06 23:02:42 +00:00
schema.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
strict_mode.py	[HOP] Reworked DispatchKey.Autograd (#151107)	2025-04-15 19:55:46 +00:00
torchbind.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
triton_kernel_wrap.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
utils.py	[HOP, map] Rework of map autograd to the new interface (#153343)	2025-08-06 23:02:42 +00:00
while_loop.py	[export] runtime asserts for while HOP subgraphs (#158467)	2025-07-23 00:34:18 +00:00
wrap.py	Add a HOP to bypass tracing of a wrapper function while tracing the wrapped function (#153487)	2025-05-22 04:24:38 +00:00