pytorch/torch/_higher_order_ops
2025-04-02 04:26:48 +00:00
..
__init__.py Rename PrimHOPBase to BaseHOP + minor changes (#146727) 2025-02-11 02:43:37 +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 [FX] Refactor immutable collections implementation (#144640) 2025-02-24 09:14:08 +00:00
associative_scan.py [hop] Rework the check of Metadata in the functionalization key (#148789) 2025-03-18 20:30:59 +00:00
auto_functionalize.py Fix auto_functionalize x inference_mode (#147925) 2025-02-26 18:05:30 +00:00
base_hop.py [invoke_subgraph] Support None in the fwd output (#150082) 2025-04-02 02:31:54 +00:00
cond.py [dynamo] add dynamo disable reasons to codebase (#150440) 2025-04-02 04:26:48 +00:00
effects.py Fix with effect lowering for list return type (#149510) 2025-03-19 19:35:08 +00:00
executorch_call_delegate.py [FX] Refactor immutable collections implementation (#144640) 2025-02-24 09:14:08 +00:00
flat_apply.py [dynamo] Make nonstrict_trace work with some pytree.register_constant-ed instances (#148007) 2025-03-05 21:28:26 +00:00
flex_attention.py Fix broken meta function for flex-attention backwards (#146563) 2025-02-08 04:13:52 +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 [hop][be] add utils for more comprehensive input alias and mutation (#145298) 2025-01-23 18:12:28 +00:00
invoke_subgraph.py [invoke_subgraph] Support None in the fwd output (#150082) 2025-04-02 02:31:54 +00:00
map.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
out_dtype.py [BE] typing for decorators - library (#138969) 2025-01-15 17:08:55 +00:00
run_const_graph.py [export] Unify single and multiple return for hops (#143227) 2025-01-13 03:31:14 +00:00
scan.py [hop] Rework the check of Metadata in the functionalization key (#148789) 2025-03-18 20:30:59 +00:00
strict_mode.py [Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117) 2024-10-09 02:29:40 +00:00
torchbind.py [reland] Support torchbind in OSS proxy executor (#150196) 2025-03-29 03:36:55 +00:00
triton_kernel_wrap.py [pytorch][triton] Warp specialization support in TritonTemplate for torchinductor (#148503) (#150122) 2025-03-29 03:36:50 +00:00
utils.py [invoke_subgraph] Support unbacked (#149298) 2025-03-31 17:25:09 +00:00
while_loop.py [hop] Rework the check of Metadata in the functionalization key (#148789) 2025-03-18 20:30:59 +00:00
wrap.py Require that all HOPs be imported at import torch time (#145939) 2025-01-29 22:27:52 +00:00