| .. |
|
__init__.py
|
[foreach_map] Initial foreach map HOP impl for inference (#142098)
|
2024-12-11 21:32:11 +00:00 |
|
associative_scan.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 |
|
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
|
[hop][be] add utils for more comprehensive input alias and mutation (#145298)
|
2025-01-23 18:12:28 +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
|
[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 |
|
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
|
[hop][be] add utils for more comprehensive input alias and mutation (#145298)
|
2025-01-23 18:12:28 +00:00 |
|
strict_mode.py
|
|
|
|
torchbind.py
|
Remove unused Python variables in torch/[_-a]* (#133492)
|
2024-12-12 17:39:14 +00:00 |
|
triton_kernel_wrap.py
|
[inductor][2/N] triton support post-#5512, user-defined triton kernels (#145348)
|
2025-01-24 00:34:01 +00:00 |
|
utils.py
|
[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)
|
2025-01-24 23:37:15 +00:00 |
|
while_loop.py
|
[hop][be] add utils for more comprehensive input alias and mutation (#145298)
|
2025-01-23 18:12:28 +00:00 |
|
wrap.py
|
|
|