Skip to content

Pull requests: pytorch/xla

New pull request New
169 Open 6,297 Closed
169 Open 6,297 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[cherry-pick to r2.6.1_aws_neuron] add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9067 opened Apr 30, 2025 by aws-yyjau Loading…
Remove trailing whitespace from the repo
#9065 opened Apr 30, 2025 by ghpvnist Loading…
Assert on empty PJRT buffers
#9062 opened Apr 29, 2025 by rpsilva-aws Loading…
add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9057 opened Apr 29, 2025 by aws-yyjau Loading…
Yifeit/pin update 0427
#9055 opened Apr 29, 2025 by tengyifei • Draft
Extend device data node binding API to not clone specified input tensors
#9054 opened Apr 29, 2025 by rpsilva-aws Loading…
1
Add clang link to PATH.
#9053 opened Apr 29, 2025 by ysiraichi Loading…
6
(Draft - PT27) Extend device data node binding to not override tensor IDs
#9052 opened Apr 28, 2025 by rpsilva-aws • Draft
Fix tensor factory when passing XLA device.
#9048 opened Apr 28, 2025 by ysiraichi • Draft
1
1
Fixes pytorch/xla#7398
#9047 opened Apr 26, 2025 by qianminj123 Loading…
2
Update pin 04/25
#9045 opened Apr 26, 2025 by bhavya01 Loading…
2
reduce redundant graph in collective op distributed SPMD and other distributed things.
#9044 opened Apr 25, 2025 by zpcore Loading…
1
[DRAFT] call_jax sets the JAX contextual mesh
#9043 opened Apr 25, 2025 by tengyifei • Draft
Update test_export_fx_passes.py
#8933 opened Apr 3, 2025 by avikchaudhuri Loading…
1
add CreateGlobalShardedData prototype
#8932 opened Apr 3, 2025 by pgmoka • Draft
1
@pgmoka
test on debian12 DO_NOT_MERGE Not for merging.
#8928 opened Apr 2, 2025 by zpcore • Draft
8
Make GPU CUDA plugin require JAX
#8919 opened Apr 1, 2025 by tengyifei • Draft
Lower upsample_bilinear2d_backward operation.
#8916 opened Apr 1, 2025 by ysiraichi • Draft
1
1
[Draft] Add Experimental limited sparse embedding bag
#8905 opened Mar 28, 2025 by amjames • Draft
1
@lsy323
1
[ragged-paged-attn] Use hidden states in kv cache and support any num…
#8891 opened Mar 26, 2025 by zpcore Loading…
[DRAFT/WIP] Add top-p masking
#8871 opened Mar 21, 2025 by hyeygit • Draft
[1/N] Initial implementation of local SPMD support distributed SPMD and other distributed things.
#8810 opened Mar 9, 2025 by lsy323 Loading…
@lsy323
39
Showcase jax.grad in torch_xla
#8800 opened Mar 5, 2025 by zpcore Loading…
6
Repro ragged paged attn kernel
#8752 opened Feb 26, 2025 by vanbasten23 • Draft
Replace setup.py with pyproject.toml
#8744 opened Feb 26, 2025 by ManfeiBai Loading…
ProTip! Exclude everything labeled bug with -label:bug.