Skip to content

Pull requests: pytorch/helion

New pull request New
93 Open 2,097 Closed
93 Open 2,097 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

Fix rope-bwd cudagraph crash and tighten gdn_fwd_h accuracy gate CLA Signed This label is managed by the Meta Open Source bot.
#2467 opened May 16, 2026 by choijon5 Contributor Loading…
[Pallas] Add test for fused_linear_jsd_fwd autograd path CLA Signed This label is managed by the Meta Open Source bot.
#2456 opened May 15, 2026 by norx1991 Contributor • Draft
[Pallas] remove jax_export_ignore_forward_compatibility setting CLA Signed This label is managed by the Meta Open Source bot.
#2452 opened May 15, 2026 by cota Collaborator Loading…
3
Fused all_gather_fp8_gemm kernel implementation CLA Signed This label is managed by the Meta Open Source bot.
#2436 opened May 15, 2026 by LironKesem • Draft
1
[No Commit] Profiling jagged gdpa on TPU CLA Signed This label is managed by the Meta Open Source bot.
#2435 opened May 15, 2026 by AmesingFlank Contributor • Draft
[WIP] fp8 Scaled mm in cute CLA Signed This label is managed by the Meta Open Source bot.
#2430 opened May 14, 2026 by yushangdi • Draft
[Autotuning] Use matmul heuristics as default seeds CLA Signed This label is managed by the Meta Open Source bot.
#2428 opened May 14, 2026 by choijon5 Contributor • Draft
Add safe dynamic autotune blacklist CLA Signed This label is managed by the Meta Open Source bot.
#2427 opened May 14, 2026 by choijon5 Contributor • Draft
A jagged_gdpa example that works on Pallas TPU CLA Signed This label is managed by the Meta Open Source bot.
#2425 opened May 14, 2026 by AmesingFlank Contributor Loading…
{experimental stack not ready for review}[cute] add fp8 dtype mapping and tcgen05 MmaF8F6F4Op capability probe CLA Signed This label is managed by the Meta Open Source bot.
#2423 opened May 14, 2026 by yushangdi • Draft
1
[compile] Pin addmm/baddbmm output dtype to self in lowering CLA Signed This label is managed by the Meta Open Source bot.
#2420 opened May 13, 2026 by karthickai Contributor • Draft
[Pallas] jsd if output arity fix CLA Signed This label is managed by the Meta Open Source bot.
#2399 opened May 12, 2026 by thcmbs Collaborator Loading…
[Pallas] Slice store values to match clamped Pallas BlockSpec ref shape CLA Signed This label is managed by the Meta Open Source bot.
#2398 opened May 12, 2026 by thcmbs Collaborator Loading…
[Pallas] Attention perf: further reduce spillage from pre-loading Q, by loading Q in-loop and not pipelining it CLA Signed This label is managed by the Meta Open Source bot.
#2397 opened May 11, 2026 by AmesingFlank Contributor • Draft
1
Stop overriding range_num_stages and range_unroll_factor for CUDA IMA workarounds CLA Signed This label is managed by the Meta Open Source bot.
#2394 opened May 11, 2026 by choijon5 Contributor • Draft
[autotune] Add observed heuristic seeds CLA Signed This label is managed by the Meta Open Source bot.
#2378 opened May 10, 2026 by choijon5 Contributor • Draft
[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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.