Thomas Bohnstingl 87cc126457 [associative_scan] partial gradient support (#162388) ...

This PR tests the partial gradient support of the `associative_scan` operation. It replaces https://github.com/bohnstingl/pytorch/pull/6

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162388
Approved by: https://github.com/ydwu4

2025-09-09 23:52:29 +00:00

..

__init__.py

[while_loop][autograd] add hop while_loop_stack_output (#160467)

2025-09-06 21:26:33 +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

[associative_scan] partial gradient support (#162388)

2025-09-09 23:52:29 +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

[map] filter none gradients and add autograd inductor tests (#160548)

2025-08-15 20:13:12 +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

Add return-max-scores to flex-attention (#161667)

2025-09-08 22:44:48 +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][hops] Remove const outputs from the speculated subgraph (#161355)

2025-09-04 18:52:01 +00:00

map.py

[map] filter none gradients and add autograd inductor tests (#160548)

2025-08-15 20:13:12 +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

[associative_scan] Autograd separated (#139939)

2025-09-08 23:30:11 +00:00

schema.py

[cond] support gen_schema for cond (#154193)

2025-08-15 17:28:13 +00:00

strict_mode.py

typing debugging.py (#160364)

2025-08-15 02:09:31 +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

Support Triton kernels in SAC region (#161541)

2025-08-28 21:15:25 +00:00

utils.py

[associative_scan] Autograd separated (#139939)

2025-09-08 23:30:11 +00:00

while_loop.py

[while_loop][autograd] support autograd_key of while_loop (#160483)

2025-09-07 21:55:29 +00:00

wrap.py

[export] Support AC HOP in pre-dispatch (#161479)

2025-08-28 01:46:25 +00:00