-
Notifications
You must be signed in to change notification settings - Fork 144
Pull requests: pytorch/helion
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Benchmarking/CI] Prevent hangs in benchmark phase via subprocess + per-config run timeout
CLA Signed
This label is managed by the Meta Open Source bot.
#2111
opened Apr 25, 2026 by
choijon5
Contributor
Loading…
[dashboard] re-enable --existing-url and filter platforms from dispatch workflow
CLA Signed
This label is managed by the Meta Open Source bot.
#2110
opened Apr 25, 2026 by
choijon5
Contributor
Loading…
Pin tritonbench commit in a file.
CLA Signed
This label is managed by the Meta Open Source bot.
#2106
opened Apr 24, 2026 by
umechand-amd
Collaborator
Loading…
[Pallas] Extend padding to matmul K-dim for fori_loop DMA and emit_pipeline
CLA Signed
This label is managed by the Meta Open Source bot.
[Pallas] Host-side padding for non-divisible pl.ds() dimensions
CLA Signed
This label is managed by the Meta Open Source bot.
#2104
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
•
Draft
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…
Fix flaky distributed and torch.compile tests
CLA Signed
This label is managed by the Meta Open Source bot.
#2098
opened Apr 24, 2026 by
choijon5
Contributor
Loading…
[Pallas] Per-tensor pipelining decision in fori_loop and emit_pipeline
CLA Signed
This label is managed by the Meta Open Source bot.
[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…
Add This label is managed by the Meta Open Source bot.
helion.from_cache() for FiniteSearch warm-start
CLA Signed
#2079
opened Apr 22, 2026 by
fulvius31
Collaborator
Loading…
[Pallas] Update xfail reasons for examples to reflect current errors
CLA Signed
This label is managed by the Meta Open Source bot.
[Pallas] Zero-pad matmul operands for non-divisible reduction tiles
CLA Signed
This label is managed by the Meta Open Source bot.
[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…
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
[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
Contributor
Loading…
[Pallas] Implement indirect gather via exact one-hot matmul
CLA Signed
This label is managed by the Meta Open Source bot.
[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…
[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…
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…
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…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.