| .. |
|
__init__.py
|
Require that all HOPs be imported at import torch time (#145939)
|
2025-01-29 22:27:52 +00:00 |
|
associative_scan.py
|
Require that all HOPs be imported at import torch time (#145939)
|
2025-01-29 22:27:52 +00:00 |
|
auto_functionalize.py
|
PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)
|
2025-01-20 22:37:26 +00:00 |
|
cond.py
|
[cond] remove warning for unsupported tuple returns (#145766)
|
2025-01-28 03:13:36 +00:00 |
|
effects.py
|
PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)
|
2025-01-20 22:37:26 +00:00 |
|
executorch_call_delegate.py
|
|
|
|
flex_attention.py
|
[hop][be] add utils for more comprehensive input alias and mutation (#145298)
|
2025-01-23 18:12:28 +00:00 |
|
foreach_map.py
|
PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)
|
2025-01-20 22:37:26 +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
|
PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)
|
2025-01-20 22:37:26 +00:00 |
|
map.py
|
|
|
|
out_dtype.py
|
[BE] typing for decorators - library (#138969)
|
2025-01-15 17:08:55 +00:00 |
|
prim_hop_base.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
|
[scan] scan dim handling in user-facing scan() (#145179)
|
2025-01-30 21:09:07 +00:00 |
|
strict_mode.py
|
|
|
|
torchbind.py
|
|
|
|
triton_kernel_wrap.py
|
[inductor] Make triton kernel autotune config defaults backward-compatible (#145494)
|
2025-01-29 00:31:39 +00:00 |
|
utils.py
|
[while_loop] specialize when cond_fn return constants (#144515)
|
2025-01-30 19:02:34 +00:00 |
|
while_loop.py
|
[hop] fix unbacked_bindings meta for while_loop (#143559)
|
2025-01-30 21:33:09 +00:00 |
|
wrap.py
|
Require that all HOPs be imported at import torch time (#145939)
|
2025-01-29 22:27:52 +00:00 |
7e7341bddd