pytorch/torch/_higher_order_ops
2024-05-31 19:47:24 +00:00
..
__init__.py Change templated_attention -> flex_attention (#125251) 2024-05-01 01:08:48 +00:00
associative_scan.py Revert "[BE] wrap deprecated function/class with typing_extensions.deprecated (#126898)" 2024-05-31 19:47:24 +00:00
auto_functionalize.py Rename impl_abstract to register_fake, part 1/2 (#123937) 2024-04-17 12:46:01 +00:00
cond.py Support trace_subgraph in _MakefxTracer (#125363) 2024-05-15 03:12:24 +00:00
effects.py [torchbind] Add inductor support (#123709) 2024-05-13 18:18:17 +00:00
flex_attention.py Add Lowering for FlexAttention Backwards (#125515) 2024-05-17 00:41:55 +00:00
map.py Support trace_subgraph in _MakefxTracer (#125363) 2024-05-15 03:12:24 +00:00
out_dtype.py
strict_mode.py Add support for multiple flexattention calls in a single compile (#125516) 2024-05-07 21:37:37 +00:00
torchbind.py Add torch._library.register_fake_class to fakify torchBind class (#122622) 2024-04-02 23:52:17 +00:00
triton_kernel_wrap.py [User-Written Triton] Handle the scf.for and scf.while case (#127065) 2024-05-24 21:01:13 +00:00
utils.py Support trace_subgraph in _MakefxTracer (#125363) 2024-05-15 03:12:24 +00:00
while_loop.py Support trace_subgraph in _MakefxTracer (#125363) 2024-05-15 03:12:24 +00:00
wrap.py