Skip to content

Pull requests: pytorch/helion

New pull request New
117 Open 2,218 Closed
117 Open 2,218 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 atomic add semantics (when targeting an output ref) CLA Signed This label is managed by the Meta Open Source bot.
#2615 opened May 27, 2026 by thcmbs Collaborator • Draft
1
[TPU][Pallas] lower hl.jagged_tile via a per-item DMA template CLA Signed This label is managed by the Meta Open Source bot.
#2614 opened May 27, 2026 by yarongmu-google Collaborator Loading…
[cute] Move promotion function to cute/device_ir_lowering CLA Signed This label is managed by the Meta Open Source bot.
#2613 opened May 27, 2026 by hinriksnaer Collaborator • Draft
[cute] Introduce DeviceIRLowering pipeline and migrate cute CLA Signed This label is managed by the Meta Open Source bot.
#2612 opened May 27, 2026 by hinriksnaer Collaborator • Draft
[fast-launcher] Concrete-tensor fast path skips _hashable_dims (Python-only) CLA Signed This label is managed by the Meta Open Source bot.
#2611 opened May 27, 2026 by yushangdi Contributor • Draft
[fast-launcher] Minimal Rust CompiledLauncher with __call__ (Chunk E) CLA Signed This label is managed by the Meta Open Source bot.
#2609 opened May 27, 2026 by yushangdi Contributor • Draft
[fast-launcher] Minimal C CompiledLauncher with tp_call (Chunk E) CLA Signed This label is managed by the Meta Open Source bot.
#2608 opened May 27, 2026 by yushangdi Contributor • Draft
[fast-launcher] Add helion._native optional Rust extension shim CLA Signed This label is managed by the Meta Open Source bot.
#2606 opened May 27, 2026 by yushangdi Contributor • Draft
[fast-launcher] Codegen rewrite to wire the pool into generated wrappers (Python-only) CLA Signed This label is managed by the Meta Open Source bot.
#2605 opened May 27, 2026 by yushangdi Contributor • Draft
[fast-launcher] Opt-in output-tensor pool (Chunk D, Python-only) CLA Signed This label is managed by the Meta Open Source bot.
#2604 opened May 27, 2026 by yushangdi Contributor • Draft
[compile] Migrate Types to a dedicated module CLA Signed This label is managed by the Meta Open Source bot.
#2603 opened May 27, 2026 by hinriksnaer Collaborator Loading…
2
examples/fp8_attention: enable for Pallas backend CLA Signed This label is managed by the Meta Open Source bot.
#2601 opened May 27, 2026 by cota Collaborator Loading…
[pallas] fix fori_loop's output mapping CLA Signed This label is managed by the Meta Open Source bot.
#2600 opened May 27, 2026 by cota Collaborator Loading…
[pallas] fix emit_pipeline's output mapping and indexing CLA Signed This label is managed by the Meta Open Source bot.
#2599 opened May 27, 2026 by cota Collaborator Loading…
5
[pallas] direct-call hot-path squeezes: sig-lock + full_invoke closure baking CLA Signed This label is managed by the Meta Open Source bot.
#2597 opened May 27, 2026 by choijon5 Contributor • Draft
[Pallas] examples: add TPU-optimized jagged_sum CLA Signed This label is managed by the Meta Open Source bot.
#2596 opened May 27, 2026 by yarongmu-google Collaborator • Draft
[pallas] cache static-shape output meta-tensor allocation in pallas launcher CLA Signed This label is managed by the Meta Open Source bot.
#2595 opened May 26, 2026 by choijon5 Contributor • Draft
[pallas] bypass JaxCallable for static-shape kernels via direct call_custom_kernel CLA Signed This label is managed by the Meta Open Source bot.
#2594 opened May 26, 2026 by choijon5 Contributor • Draft
[pallas] fast-path host-side launcher with pre-computed cache entries CLA Signed This label is managed by the Meta Open Source bot.
#2593 opened May 26, 2026 by choijon5 Contributor • Draft
[pallas] launcher infrastructure: VMEM strip BlockSpecs + reduction-axis kwargs CLA Signed This label is managed by the Meta Open Source bot.
#2592 opened May 26, 2026 by choijon5 Contributor • Draft
Fix fbcode CI torch.compile fusion with newer PyTorch CLA Signed This label is managed by the Meta Open Source bot.
#2567 opened May 23, 2026 by choijon5 Contributor Loading…
3
Speed up Helion kernel launches by avoiding repeated Python work CLA Signed This label is managed by the Meta Open Source bot.
#2565 opened May 23, 2026 by yushangdi Contributor • Draft
2
[Pallas] Fix layernorm example tolerances and split bwd test CLA Signed This label is managed by the Meta Open Source bot.
#2560 opened May 22, 2026 by thcmbs Collaborator • Draft
[Pallas] Propagate inner tile alignment min_size to bounding outer tiles CLA Signed This label is managed by the Meta Open Source bot.
#2559 opened May 22, 2026 by thcmbs Collaborator Loading…
Skip even more Python on repeated identical calls CLA Signed This label is managed by the Meta Open Source bot.
#2537 opened May 20, 2026 by choijon5 Contributor • Draft
ProTip! Exclude everything labeled bug with -label:bug.