Skip to content

Pull requests: pytorch/helion

New pull request New
77 Open 1,522 Closed
77 Open 1,522 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

[Pallas] Fix multi-dim padding overwriting original tensor reference CLA Signed This label is managed by the Meta Open Source bot.
#2120 opened Apr 27, 2026 by norx1991 Contributor Loading…
[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] Add pl.multiple_of alignment hint to pl.ds() offsets CLA Signed This label is managed by the Meta Open Source bot.
#2116 opened Apr 27, 2026 by norx1991 Contributor • Draft
[Pallas] Extend padding to fori_loop DMA and emit_pipeline via _record_pad_info CLA Signed This label is managed by the Meta Open Source bot.
#2105 opened Apr 24, 2026 by norx1991 Contributor Loading…
[Pallas] Adding a tunable pre_broadcast optimization pass for TPU scratch buffers, improving TPU attention perf CLA Signed This label is managed by the Meta Open Source bot.
#2103 opened Apr 24, 2026 by AmesingFlank Contributor Loading…
4
autotuner: cap tile size for imbalanced 2D grid dims CLA Signed This label is managed by the Meta Open Source bot.
#2102 opened Apr 24, 2026 by umechand-amd Collaborator Loading…
4
[Pallas] Per-tensor pipelining decision in fori_loop and emit_pipeline CLA Signed This label is managed by the Meta Open Source bot.
#2093 opened Apr 23, 2026 by norx1991 Contributor • Draft
[TPU][Pallas] Add automatic compiler padding for unevenly tiled loops CLA Signed This label is managed by the Meta Open Source bot.
#2085 opened Apr 23, 2026 by yarongmu-google Collaborator Loading…
7
Add helion.from_cache() for FiniteSearch warm-start CLA Signed This label is managed by the Meta Open Source bot.
#2079 opened Apr 22, 2026 by fulvius31 Collaborator Loading…
3
[Pallas] Update xfail reasons for examples to reflect current errors CLA Signed This label is managed by the Meta Open Source bot.
#2074 opened Apr 22, 2026 by norx1991 Contributor • Draft
[Pallas] Zero-pad matmul operands for non-divisible reduction tiles CLA Signed This label is managed by the Meta Open Source bot.
#2062 opened Apr 20, 2026 by norx1991 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
[Pallas] Indirect gather with pluggable strategies CLA Signed This label is managed by the Meta Open Source bot.
#2054 opened Apr 20, 2026 by thcmbs Collaborator Loading…
9
[Pallas] Implement indirect gather via exact one-hot matmul CLA Signed This label is managed by the Meta Open Source bot.
#2035 opened Apr 17, 2026 by thcmbs Collaborator • Draft
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 Loading…
5
[Autotuner] Apply numel constraints on search neighbors CLA Signed This label is managed by the Meta Open Source bot.
#2011 opened Apr 13, 2026 by fulvius31 Collaborator Loading…
2
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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.