Skip to content

Pull requests: pytorch/xla

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

Update tuned block table and add benchmarking/autotune scripts for quantized matmul pallas kernel.
#9322 opened Jun 10, 2025 by vanbasten23 • Draft
Update submodules whenever we switch branches.
#9320 opened Jun 9, 2025 by zhanyong-wan Loading…
Use the actual vmem usage in pallas call
#9316 opened Jun 9, 2025 by Chenyaaang Loading…
1
Fix basic_training.py
#9313 opened Jun 9, 2025 by zzzwen Loading…
6
Fix race condition in runAtenTest
#9306 opened Jun 6, 2025 by benawilson Loading…
1
Implement prng_key as a mutable array
#9305 opened Jun 6, 2025 by tengyifei Loading…
1
3
Add an option to not use dlpack.
#9304 opened Jun 6, 2025 by qihqi Loading…
3
cleanup: remove defunct --distinct_host_configuration flag
#9300 opened Jun 5, 2025 by rickeylev Loading…
1
Add cache support for scan_layers
#9297 opened Jun 5, 2025 by iwknow Loading…
1
Add DLRM DCN v2 with JAX Sparsecore
#9287 opened Jun 3, 2025 by chandrasekhard2 Loading…
1
WIP - do not review; Support python 3.12
#9274 opened Jun 1, 2025 by zhanyong-wan • Draft
1
doc: update pytorch-on-xla-devices and troubleshoot doc for tensor synchronization issue
#9258 opened May 28, 2025 by aws-yyjau Loading…
34
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
2
[torchax]: JittableModule statedict handling
#9195 opened May 19, 2025 by zmelumian972 Loading…
9
Migrate to correct logger interface
#9191 opened May 17, 2025 by emmanuel-ferdman Loading…
2
Switch bazel to default to Clang for C/C++ compilation
#9169 opened May 14, 2025 by zhanyong-wan • Draft
1
[Test] Add torch pin 152601
#9167 opened May 14, 2025 by bhavya01 • Draft
(Do not submit) Set StableHLO Compilation to true for CI
#9141 opened May 12, 2025 by GleasonK • Draft
Lower clamp
#9123 opened May 8, 2025 by ghpvnist • Draft
Update numerical verification for SPMD Linear checkpointing CI CI related change
#9113 opened May 7, 2025 by sdasgup3 Loading…
1
Collect coverage data. DO_NOT_MERGE Not for merging. DO_NOT_REVIEW_YET
#9107 opened May 7, 2025 by ysiraichi • Draft
Move requires_jax to inner flash_attention functions
#9098 opened May 5, 2025 by bhavya01 Loading…
1
1
Add features needed for vllm
#9092 opened May 5, 2025 by wenxindongwork • Draft
[not for review] update pin
#9087 opened May 4, 2025 by bobrenjc93 Loading…
6
Check the scan op input for requires_grad
#9083 opened May 2, 2025 by haifeng-jin Loading…
1
@haifeng-jin
1
ProTip! Exclude everything labeled bug with -label:bug.