Skip to content

Pull requests: pytorch/helion

New pull request New
126 Open 2,119 Closed
126 Open 2,119 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[runtime:pallas] migrate to torch_tpu's new Pallas buffer donation API CLA Signed This label is managed by the Meta Open Source bot.
#2351 opened May 7, 2026 by cota Collaborator Loading…
2
WIP Pallas atomic all launchers v2 CLA Signed This label is managed by the Meta Open Source bot.
#2346 opened May 7, 2026 by thcmbs Collaborator • Draft
Add multi tile loop support to autodiff CLA Signed This label is managed by the Meta Open Source bot.
#2338 opened May 7, 2026 by karthickai Contributor • Draft
[Pallas] Skip factory tensor padding for Pallas backend CLA Signed This label is managed by the Meta Open Source bot.
#2279 opened May 5, 2026 by norx1991 Contributor • Draft
[TPU][Pallas] relax tolerances and fix Pallas autotuning OOM in layer_norm CLA Signed This label is managed by the Meta Open Source bot.
#2272 opened May 5, 2026 by yarongmu-google Collaborator • Draft
A jagged_hstu_attention example that works on Pallas TPU CLA Signed This label is managed by the Meta Open Source bot.
#2218 opened May 3, 2026 by AmesingFlank Contributor Loading…
[Pallas] Use LONG_INT_TYPE for jagged offsets in examples and tests CLA Signed This label is managed by the Meta Open Source bot.
#2132 opened Apr 28, 2026 by norx1991 Contributor • Draft
[Autotuner] Raise default min_improvement_delta to 0.003 CLA Signed This label is managed by the Meta Open Source bot.
#2119 opened Apr 27, 2026 by choijon5 Contributor • Draft
[Pallas] Switch gather to jnp.take_along_axis (for JAX issue filing) CLA Signed This label is managed by the Meta Open Source bot.
#2061 opened Apr 20, 2026 by AmesingFlank Contributor • Draft
[Pallas] Lower aten gather using one_hot + sum for TPU compatibility, unblocking cross_entropy CLA Signed This label is managed by the Meta Open Source bot.
#2060 opened Apr 20, 2026 by AmesingFlank Contributor Loading…
2
Use torch.gather instead of generic int indexing for cross_entropy example CLA Signed This label is managed by the Meta Open Source bot.
#2058 opened Apr 20, 2026 by AmesingFlank Contributor • Draft
2
CI: retry flaky CUDA compute check CLA Signed This label is managed by the Meta Open Source bot.
#2034 opened Apr 17, 2026 by choijon5 Contributor • Draft
[TPU][Pallas]Fix example/cross_entropy.py on Pallas TPU CLA Signed This label is managed by the Meta Open Source bot.
#2019 opened Apr 14, 2026 by yarongmu-google Collaborator • Draft
9
WIP: fp8 all gather matmul CLA Signed This label is managed by the Meta Open Source bot.
#1974 opened Apr 7, 2026 by shunting314 Contributor Loading…
warning for process group name not found conditionally CLA Signed This label is managed by the Meta Open Source bot.
#1973 opened Apr 7, 2026 by shunting314 Contributor Loading…
Use persistent+tensor_descriptor defaults for dot kernels on Blackwell (sm100+) CLA Signed This label is managed by the Meta Open Source bot.
#1964 opened Apr 7, 2026 by choijon5 Contributor Loading…
2
fix _clone_args when there are non-tensor input CLA Signed This label is managed by the Meta Open Source bot.
#1963 opened Apr 6, 2026 by shunting314 Contributor Loading…
3
autotune fp8 matmul reduce scatter in unit test CLA Signed This label is managed by the Meta Open Source bot.
#1962 opened Apr 6, 2026 by shunting314 Contributor Loading…
only check accuracy of mutated args CLA Signed This label is managed by the Meta Open Source bot.
#1961 opened Apr 6, 2026 by shunting314 Contributor Loading…
3
[compiler] add basic lambda support CLA Signed This label is managed by the Meta Open Source bot.
#1958 opened Apr 6, 2026 by hinriksnaer Collaborator Loading…
1
Raise error when autotune_with_torch_compile_fusion is set without torch_compile_fusion CLA Signed This label is managed by the Meta Open Source bot.
#1951 opened Apr 4, 2026 by yf225 Contributor Loading…
3 tasks
1
[Auto-Recovery] Add checkpoint save/load/resume CLA Signed This label is managed by the Meta Open Source bot.
#1947 opened Apr 4, 2026 by yf225 Contributor Loading…
2
[Auto-Recovery] Refactor search algorithms for checkpointable state CLA Signed This label is managed by the Meta Open Source bot.
#1946 opened Apr 4, 2026 by yf225 Contributor Loading…
3
[Not finished] Linear attention examples CLA Signed This label is managed by the Meta Open Source bot.
#1941 opened Apr 3, 2026 by v0i0 Contributor • Draft
1
actually make pallas interpret work in ci CLA Signed This label is managed by the Meta Open Source bot.
#1938 opened Apr 2, 2026 by v0i0 Contributor Loading…
1
ProTip! Exclude everything labeled bug with -label:bug.