Skip to content

Pull requests: pytorch/helion

New pull request New
89 Open 1,583 Closed
89 Open 1,583 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

[TPU] [Pallas] Fix Pallas codegen for _if node to handle lax.cond return values CLA Signed This label is managed by the Meta Open Source bot.
#1939 opened Apr 3, 2026 by yarongmu-google • 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 Loading…
[Pallas] Use jnp.where() for branches with tensor-derived conditions but no in-place tensor updates CLA Signed This label is managed by the Meta Open Source bot.
#1936 opened Apr 2, 2026 by AmesingFlank Loading…
Refactor if-else branches, unblock test_if_new_variable_in_static_range for Pallas TPU CLA Signed This label is managed by the Meta Open Source bot.
#1935 opened Apr 2, 2026 by AmesingFlank Loading…
[Autotuner] Add crash recovery script for unrecoverable CUDA errors CLA Signed This label is managed by the Meta Open Source bot.
#1934 opened Apr 2, 2026 by yf225 Loading…
use is_symm_mem_tensor if the API is available CLA Signed This label is managed by the Meta Open Source bot.
#1933 opened Apr 2, 2026 by shunting314 Loading…
Upgrade MI325 CI from ROCm 7.0 to 7.2 ciflow/rocm-mi300 CLA Signed This label is managed by the Meta Open Source bot. module: rocm
#1931 opened Apr 2, 2026 by v0i0 Loading…
1 task
1
[autotuner] Introduce BenchmarkProvider abstraction for kernel benchmarking CLA Signed This label is managed by the Meta Open Source bot.
#1928 opened Apr 2, 2026 by hinriksnaer Loading…
[Pallas] Use fori_loop if loop bounds includes non-constexpr symbolic values CLA Signed This label is managed by the Meta Open Source bot.
#1927 opened Apr 2, 2026 by AmesingFlank Loading…
3
[autotune] Decouple PrecompileFuture from BaseSearch via PrecompileContext CLA Signed This label is managed by the Meta Open Source bot.
#1925 opened Apr 2, 2026 by hinriksnaer Loading…
[Autotuner] Add crash recovery script for unrecoverable CUDA errors CLA Signed This label is managed by the Meta Open Source bot.
#1923 opened Apr 2, 2026 by yf225 Loading…
[Autotuner] Auto-checkpoint feature and ability to resume from checkpoint CLA Signed This label is managed by the Meta Open Source bot.
#1920 opened Apr 2, 2026 by yf225 Loading…
1
[Pallas] Fix fori_loop multi-dim index decomposition CLA Signed This label is managed by the Meta Open Source bot.
#1917 opened Apr 2, 2026 by norx1991 Loading…
2
[Pallas] Fix emit_pipeline program_id mapping with loop_order reordering CLA Signed This label is managed by the Meta Open Source bot.
#1916 opened Apr 2, 2026 by norx1991 Loading…
1
[Pallas] Register pallas_loop_type only when inner loops exist CLA Signed This label is managed by the Meta Open Source bot.
#1915 opened Apr 2, 2026 by norx1991 Loading…
Add HELION_AUTOTUNE_PRECOMPILE_REPS setting to run kernel multiple times during spawn precompile CLA Signed This label is managed by the Meta Open Source bot. fb-exported meta-exported
#1914 opened Apr 1, 2026 by yf225 Loading…
1
[DO NOT REVIEW] [Pallas] Add TPU nightly benchmark workflow and runner CLA Signed This label is managed by the Meta Open Source bot.
#1913 opened Apr 1, 2026 by norx1991 • Draft
3 of 5 tasks
1
dbg CLA Signed This label is managed by the Meta Open Source bot.
#1911 opened Apr 1, 2026 by shunting314 Loading…
WIP: consolidate accuracy check APIs CLA Signed This label is managed by the Meta Open Source bot.
#1910 opened Apr 1, 2026 by shunting314 Loading…
Enable autotuning for layernorm CLA Signed This label is managed by the Meta Open Source bot. fb-exported meta-exported
#1908 opened Apr 1, 2026 by Hamlin-Li Loading…
1
Epilogue subtiling: store indexing fix, example, and tuple output support in run_example CLA Signed This label is managed by the Meta Open Source bot.
#1907 opened Apr 1, 2026 by choijon5 Loading…
4
[autotuner] Work around Triton two-dot shared-memory IMA on H100 CLA Signed This label is managed by the Meta Open Source bot.
#1905 opened Apr 1, 2026 by yf225 Loading…
3 tasks done
1
Add get_block_size_origin helper CLA Signed This label is managed by the Meta Open Source bot.
#1898 opened Mar 31, 2026 by jansel Loading…
torch_tpu: move pin to c889f5404 CLA Signed This label is managed by the Meta Open Source bot.
#1895 opened Mar 31, 2026 by cota • Draft
[DO NOT MERGE] Test about XPU CI CLA Signed This label is managed by the Meta Open Source bot.
#1894 opened Mar 31, 2026 by Stonepia • Draft
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.