pytorch/torch/_higher_order_ops
2025-09-27 18:13:15 +00:00
..
__init__.py [dynamo][hop] Introduce Local Map HOP (#161458) 2025-09-17 09:32: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 [hop] refactor check input alias and mutation to be a graph pass (#162025) 2025-09-27 18:13:15 +00:00
auto_functionalize.py [doc]: Small typos (#162982) 2025-09-16 17:42:19 +00:00
base_hop.py [hop] refactor check input alias and mutation to be a graph pass (#162025) 2025-09-27 18:13:15 +00:00
cond.py [hop] refactor check input alias and mutation to be a graph pass (#162025) 2025-09-27 18:13:15 +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 [Flex] Fix silent correctness w/ backpropping grads (#163677) 2025-09-24 02:12:19 +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 [hop] refactor check input alias and mutation to be a graph pass (#162025) 2025-09-27 18:13:15 +00:00
local_map.py [hop] support local_map + SAC (#163322) 2025-09-24 04:57:40 +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
partitioner.py [scan] create fw and bw graphs via partitioning (#162754) 2025-09-27 18:13:15 +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 [scan] create fw and bw graphs via partitioning (#162754) 2025-09-27 18:13:15 +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 Fix specialize_impl from triton.runtime.jit (#163844) 2025-09-26 06:37:26 +00:00
utils.py [scan] create fw and bw graphs via partitioning (#162754) 2025-09-27 18:13:15 +00:00
while_loop.py [hop] refactor check input alias and mutation to be a graph pass (#162025) 2025-09-27 18:13:15 +00:00
wrap.py [export] Fix wrap_with_set_grad_enabled retracing (#163295) 2025-09-21 22:54:40 +00:00