Skip to content

Pull requests: pytorch/xla

New pull request New
183 Open 6,575 Closed
183 Open 6,575 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

Dump C++ and Status propagation stacktraces.
#9492 opened Jul 19, 2025 by ysiraichi • Draft
Misc fixes:
#9491 opened Jul 18, 2025 by qihqi Loading…
Support SPMD placeholder tensors
#9489 opened Jul 17, 2025 by rpsilva-aws Loading…
Implement XLAShardedTensor._spec and test
#9488 opened Jul 17, 2025 by aws-cph Loading…
9
Update torchprime tests to Python 3.12 bug Something isn't working testing Testing and coverage related issues.
#9481 opened Jul 15, 2025 by pgmoka • Draft
@pgmoka
feat: abstraction of xla::OpSharding proto using wrapper class
#9467 opened Jul 10, 2025 by kvshbg-aws Loading…
Introduce multi-operand collective permute
#9450 opened Jul 8, 2025 by rpsilva-aws Loading…
Error Handling: refactor ExecuteComputation and ExecuteReplicated to propagate status.
#9445 opened Jul 3, 2025 by ysiraichi • Draft
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…
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 Loading…
3
Add a test on OOM error
#9414 opened Jun 26, 2025 by zhanyong-wan • 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
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
Clarify xr.device_type() API and use them
#9343 opened Jun 11, 2025 by ghpvnist Loading…
@ghpvnist
wip: use torch from a wheel
#9340 opened Jun 11, 2025 by rickeylev • Draft
1 task
5
Test making this work from forks
#9335 opened Jun 11, 2025 by tengyifei • Draft
Do not inherit secrets
#9331 opened Jun 10, 2025 by tengyifei • Draft
Update tuned block table and add benchmarking/autotune scripts for quantized matmul pallas kernel.
#9322 opened Jun 10, 2025 by vanbasten23 • Draft
Add DLRM DCN v2 with JAX Sparsecore
#9287 opened Jun 3, 2025 by chandrasekhard2 Loading…
2
WIP - do not review; Support python 3.12
#9274 opened Jun 1, 2025 by zhanyong-wan • Draft
1
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
6
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.