Skip to content

Pull requests: pytorch/xla

New pull request New
185 Open 6,416 Closed
185 Open 6,416 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

Remove a few CUDA tests in preparation for #9202
#9286 opened Jun 3, 2025 by ghpvnist • Draft
@ghpvnist
Deprecate devkind field from xla_model.xla_device
#9284 opened Jun 3, 2025 by ghpvnist • Draft
1
@ghpvnist
[WIP] Support fp8 lowering in scalar
#9283 opened Jun 3, 2025 by yaochengji • Draft
Do not clear UncachedCompile between graph executions
#9282 opened Jun 2, 2025 by haifeng-jin • Draft
1
@haifeng-jin
1
Split training tests to separate test shard
#9281 opened Jun 2, 2025 by ghpvnist Loading…
@ghpvnist
2
Add w8a8 quantized matmul kernel
#9278 opened Jun 2, 2025 by vanbasten23 Loading…
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…
33
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
2
[torchax] Fixes test for kthvalue pytorch/xla#7458
#9223 opened May 21, 2025 by qianminj123 Loading…
1
Disable AMP by default on CPU
#9218 opened May 20, 2025 by haifeng-jin Loading…
1
@haifeng-jin
7
Introduce annotate_custom_sharding binding
#9203 opened May 19, 2025 by rpsilva-aws Loading…
13
[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
Add cache to value_and_grad_partitioned
#9163 opened May 14, 2025 by iwknow Loading…
9
Test torchprime from PyTorch/XLA
#9152 opened May 13, 2025 by tengyifei Loading…
1
8
(Do not submit) Set StableHLO Compilation to true for CI
#9141 opened May 12, 2025 by GleasonK • Draft
remove version number from nightly build references
#9130 opened May 9, 2025 by bfolie Loading…
4
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.