Skip to content

Pull requests: pytorch/helion

New pull request New
179 Open 2,738 Closed
179 Open 2,738 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

Remove KDA serving experiments CLA Signed This label is managed by the Meta Open Source bot.
#3214 opened Jul 31, 2026 by ethche Contributor Loading…
[pretuned] Optimize fused_qk_norm_rope on B200 CLA Signed This label is managed by the Meta Open Source bot.
#3213 opened Jul 30, 2026 by yushangdi Contributor • Draft
[CuTe] Beat CUTLASS across the scaled FP8 matmul sweep CLA Signed This label is managed by the Meta Open Source bot.
#3209 opened Jul 30, 2026 by yushangdi Contributor • Draft
4
[Pallas] Run kl_div at fp32 on the TPU bridge to match run_tpu.py CLA Signed This label is managed by the Meta Open Source bot.
#3208 opened Jul 30, 2026 by norx1991 Contributor Loading…
[Pallas] Move welford to the run_tpu.py TPU holdout CLA Signed This label is managed by the Meta Open Source bot.
#3207 opened Jul 30, 2026 by norx1991 Contributor • Draft
[Pallas] Downgrade compact_worklist grouping gracefully when kernel doesn't match CLA Signed This label is managed by the Meta Open Source bot.
#3206 opened Jul 29, 2026 by norx1991 Contributor Loading…
[Pallas] xsa TPU sweep: use bfloat16 (was float16, hit Mosaic vector-type bug) CLA Signed This label is managed by the Meta Open Source bot.
#3204 opened Jul 29, 2026 by norx1991 Contributor Loading…
[cute] refactor: add generic tcgen05 fragment epilogues instead of strict allowlist CLA Signed This label is managed by the Meta Open Source bot.
#3203 opened Jul 29, 2026 by ethche Contributor Loading…
[pallas] Support dependent Tile.end bounds in Pallas CLA Signed This label is managed by the Meta Open Source bot.
#3201 opened Jul 29, 2026 by ethche Contributor Loading…
2
torch_tpu: update pin to 2b237048ad2d20bb083bc1eebb7b73b1a4d3daa0 CLA Signed This label is managed by the Meta Open Source bot.
#3199 opened Jul 29, 2026 by cota Collaborator • Draft
Add ellipsis indexing support in normal mode CLA Signed This label is managed by the Meta Open Source bot.
#3198 opened Jul 29, 2026 by parsshar-RH Contributor Loading…
Fix numel constraints for invariant extents CLA Signed This label is managed by the Meta Open Source bot.
#3194 opened Jul 28, 2026 by ethche Contributor Loading…
add w4a4 gemm kernel to examples/nvfp4_gemm.py CLA Signed This label is managed by the Meta Open Source bot.
#3191 opened Jul 28, 2026 by LironKesem Contributor Loading…
1
Refactor the Pallas launcher to make it easier to operate on Jax Arrays as well as TorchTPU tensors CLA Signed This label is managed by the Meta Open Source bot.
#3190 opened Jul 28, 2026 by AmesingFlank Contributor Loading…
Support OutputCodeOptions(jax_fn=True) for Pallas (JAX-array entrypoints) CLA Signed This label is managed by the Meta Open Source bot.
#3186 opened Jul 28, 2026 by AmesingFlank Contributor Loading…
Support OutputCodeOptions(allow_helion_deps=False) for the Pallas backend (TorchTPU) CLA Signed This label is managed by the Meta Open Source bot.
#3184 opened Jul 28, 2026 by AmesingFlank Contributor Loading…
Add OutputCodeOptions(allow_helion_deps=False) to BoundKernel.to_code for standalone Triton kernels CLA Signed This label is managed by the Meta Open Source bot.
#3181 opened Jul 28, 2026 by AmesingFlank Contributor Loading…
16
Preserve hl.specialize values for exported kernels CLA Signed This label is managed by the Meta Open Source bot.
#3171 opened Jul 28, 2026 by yushangdi Contributor • Draft
Support grouped reshape reductions on CuTe CLA Signed This label is managed by the Meta Open Source bot.
#3170 opened Jul 27, 2026 by yushangdi Contributor • Draft
Support scalar FP8 conversion in CuTe CLA Signed This label is managed by the Meta Open Source bot.
#3169 opened Jul 27, 2026 by yushangdi Contributor Loading…
Support symbolic tile reshape constraints CLA Signed This label is managed by the Meta Open Source bot.
#3168 opened Jul 27, 2026 by yushangdi Contributor • Draft
[autotuner][cute] repair partial FFI candidates away from the seed, not onto it CLA Signed This label is managed by the Meta Open Source bot.
#3167 opened Jul 27, 2026 by calebmkim Contributor • Draft
Support in-tile reshapes for symbolic block sizes CLA Signed This label is managed by the Meta Open Source bot.
#3165 opened Jul 27, 2026 by yushangdi Contributor Loading…
1
6
Cap persistent kernel grids at logical PID count CLA Signed This label is managed by the Meta Open Source bot.
#3160 opened Jul 25, 2026 by yushangdi Contributor • Draft
[cute] Support explicit rank-1 column broadcasts CLA Signed This label is managed by the Meta Open Source bot.
#3159 opened Jul 24, 2026 by ethche Contributor Loading…
2
ProTip! Type g i on any issue or pull request to go back to the issue listing page.