Skip to content

Pull requests: pytorch/xla

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

[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper
#9180 opened May 16, 2025 by yaochengji Loading…
8
Add interop with flax (Part 1)
#9176 opened May 15, 2025 by qihqi Loading…
test: add test_xla_graph_execution to test flags (_set_allow_execution with PT_XLA_DEBUG_LEVEL)
#9171 opened May 14, 2025 by aws-yyjau Loading…
29
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…
1
wip: test
#9152 opened May 13, 2025 by tengyifei • Draft
Reflow ci.md to 80 chars
#9151 opened May 13, 2025 by tengyifei Loading…
3
Update ci.md and fix typo
#9150 opened May 13, 2025 by tengyifei Loading…
4
(Do not submit) Set StableHLO Compilation to true for CI
#9141 opened May 12, 2025 by GleasonK • Draft
In pjrt runtime client, raise a Python exception if XLA compilation fails.
#9138 opened May 10, 2025 by zhanyong-wan Loading…
Document the difference between tracing time and execution time documentation
#9133 opened May 9, 2025 by sdasgup3 Loading…
1
24
remove version number from nightly build references
#9130 opened May 9, 2025 by bfolie Loading…
3
Lower clamp
#9123 opened May 8, 2025 by ghpvnist • Draft
get master ip address for neuron device
#9120 opened May 8, 2025 by aws-zhenguo Loading…
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
9082 educate users on mat mul precision
#9103 opened May 6, 2025 by yaoshiang Loading…
1
47
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
Fix + Run DynamicShapeDetector tests on CI.
#9075 opened May 1, 2025 by ysiraichi Loading…
Replace xm.mark_step with torch_xla.sync() wherever possible
#9070 opened May 1, 2025 by ghpvnist Loading…
1
5
Assert on empty PJRT buffers
#9062 opened Apr 29, 2025 by rpsilva-aws Loading…
4
ProTip! Exclude everything labeled bug with -label:bug.