Sam Ginzburg ec1f56fdcf [user triton] add support for prune_configs_by in @triton.autotune (#142207) ...

This PR adds support for prune_configs_by in the @triton.autotune decorator [docs](https://triton-lang.org/main/python-api/generated/triton.autotune.html#triton.autotune). Supporting this lets users reduce autotuning time by running user-supplied code (early_config_prune, perf_model) to prune the provided list of configs.

We implement this by realizing args/kwargs in call_triton_kernel(...), and then calling kernel.prune_configs(...).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/142207
Approved by: https://github.com/zou3519, https://github.com/aakhundov

2025-01-04 03:50:28 +00:00

..

__init__.py

[foreach_map] Initial foreach map HOP impl for inference (#142098)

2024-12-11 21:32:11 +00:00

associative_scan.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

auto_functionalize.py

Generate slice.Tensor view operations instead of as_strided when split is used in the original program. (#137225)

2024-10-23 17:42:16 +00:00

cond.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

effects.py

Remove some unused type ignores (round 1) (#142325)

2024-12-09 18:23:46 +00:00

executorch_call_delegate.py

[hop] require hops to override __call__. (#134352)

2024-08-28 19:56:40 +00:00

flex_attention.py

[FlexAttention] Fix broken eager tracing (#143344)

2024-12-17 09:42:36 +00:00

foreach_map.py

[foreach_map] Initial foreach map HOP impl for inference (#142098)

2024-12-11 21:32:11 +00:00

hints_wrap.py

[HOO] add hints_wrapper to support passing context hints (#132860)

2024-08-26 18:21:22 +00:00

invoke_subgraph.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

map.py

[BE]: Apply PERF401 autofixes from ruff (#140980)

2024-11-20 17:52:07 +00:00

out_dtype.py

Make the __module__ name of HOO to be always "torch.ops.higher_order" (#132775)

2024-08-08 16:55:09 +00:00

prim_hop_base.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

run_const_graph.py

[hop] require hops to override __call__. (#134352)

2024-08-28 19:56:40 +00:00

scan.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +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

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

triton_kernel_wrap.py

[user triton] add support for prune_configs_by in @triton.autotune (#142207)

2025-01-04 03:50:28 +00:00

utils.py

[while_loop][dynamo] auto-unspecialize int input and output to unbacked symints (#143106)

2025-01-03 19:01:07 +00:00

while_loop.py

[while_loop][dynamo] auto-unspecialize int input and output to unbacked symints (#143106)

2025-01-03 19:01:07 +00:00

wrap.py

Allow fx graph caching higher order operators (opt-in) (#135877)

2024-09-24 17:23:09 +00:00