Skip to content

Pull requests: pytorch/xla

New pull request New
194 Open 6,551 Closed
194 Open 6,551 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

Add r2.9 release nightly tests to README
#9478 opened Jul 14, 2025 by pgmoka Loading…
@pgmoka
Fix spmd sharding visualization when device index is >= 10
#9475 opened Jul 11, 2025 by jeffhataws Loading…
1
Add dtensor mesh conversion test
#9474 opened Jul 11, 2025 by aws-cph Loading…
6
Optimize w8a8 pallas kernel
#9473 opened Jul 11, 2025 by kyuyeunk Loading…
21
feat: abstraction of xla::OpSharding proto using wrapper class
#9467 opened Jul 10, 2025 by kvshbg-aws Loading…
commit
#9453 opened Jul 8, 2025 by qihqi • Draft
Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter
#9451 opened Jul 8, 2025 by zmelumian972 Loading…
3
Introduce multi-operand collective permute
#9450 opened Jul 8, 2025 by rpsilva-aws Loading…
Fix duplicate labels and other docs build warnings
#9446 opened Jul 4, 2025 by melissawm Loading…
1
6
Error Handling: refactor ExecuteComputation and ExecuteReplicated to propagate status.
#9445 opened Jul 3, 2025 by ysiraichi • Draft
1
implement collective all_to_all op
#9442 opened Jul 2, 2025 by bfolie Loading…
7
Add WITH_LOCATION macros for propagating external library errors.
#9441 opened Jul 2, 2025 by ysiraichi Loading…
1
Convert some XLA_CHECKs to fatal errors.
#9439 opened Jul 2, 2025 by zhanyong-wan • Draft
feat: add normalize_tile_assignment function needed for local SPMD
#9436 opened Jul 1, 2025 by kvshbg-aws Loading…
Implement collective gather op
#9435 opened Jul 1, 2025 by bfolie Loading…
11
Error Handling: propagate status for ReleaseGilAndTransferData and XlaDataToTensors.
#9431 opened Jul 1, 2025 by ysiraichi • Draft
1
Error Handling: refactor ComputationClient::TransferFromDevice to propagate status.
#9429 opened Jul 1, 2025 by ysiraichi • Draft
3
Error Handling: refactor the existing ComputationClient implementations to use status QOL functions.
#9420 opened Jun 28, 2025 by ysiraichi • Draft
1
Add a test on OOM error
#9414 opened Jun 26, 2025 by zhanyong-wan • Draft
Misc changes: default sharding + allow scalar tensor math
#9398 opened Jun 22, 2025 by qihqi • Draft
[wip]ruff inter on torchax
#9396 opened Jun 20, 2025 by zzzwen Loading…
1
Fix nested stableHLO composite regions
#9385 opened Jun 18, 2025 by Carlomus Loading…
3
Torchax: Allow reuse of the jittable procedure with different functio…
#9374 opened Jun 17, 2025 by zmelumian972 Loading…
1
implement send and recv using collective_permute
#9373 opened Jun 17, 2025 by bfolie Loading…
20
[WIP] Add autocast in torchax
#9361 opened Jun 16, 2025 by qihqi • Draft
ProTip! Type g i on any issue or pull request to go back to the issue listing page.