-
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
[Pallas] Extract _classify_pipelined_tensors helper
CLA Signed
This label is managed by the Meta Open Source bot.
Small attention optimization: pre-scale q tile with qk_scale
CLA Signed
This label is managed by the Meta Open Source bot.
#2157
opened Apr 30, 2026 by
AmesingFlank
Contributor
Loading…
[HostFunction] Extract This label is managed by the Meta Open Source bot.
_parse_source from Hostfunction.__init__
CLA Signed
#2154
opened Apr 30, 2026 by
hinriksnaer
Collaborator
Loading…
Add offsets kwarg to hl.rand for explicit Philox offsets
CLA Signed
This label is managed by the Meta Open Source bot.
#2153
opened Apr 30, 2026 by
karthickai
Contributor
Loading…
[BoundKernel] added _normalize_config
CLA Signed
This label is managed by the Meta Open Source bot.
#2152
opened Apr 30, 2026 by
hinriksnaer
Collaborator
Loading…
[Pallas] Use torch.addmm in matmul_layernorm K-loop
CLA Signed
This label is managed by the Meta Open Source bot.
#2141
opened Apr 29, 2026 by
norx1991
Contributor
Loading…
use guard clause for cache
CLA Signed
This label is managed by the Meta Open Source bot.
#2133
opened Apr 29, 2026 by
hinriksnaer
Collaborator
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.
[Pallas] Enable se_block tests on TPU + simplify skipIfCudaCapabilityLessThan
CLA Signed
This label is managed by the Meta Open Source bot.
#2131
opened Apr 28, 2026 by
norx1991
Contributor
Loading…
[Autotuner] Long-lived worker pool for parallel precompile
CLA Signed
This label is managed by the Meta Open Source bot.
[Autotuner] Raise default min_improvement_delta to 0.003
CLA Signed
This label is managed by the Meta Open Source bot.
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…
[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] 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] 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…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.