Skip to content

Pull requests: pytorch/xla

New pull request New
172 Open 6,301 Closed
172 Open 6,301 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

update DTensor usage with upstream
#9079 opened May 2, 2025 by zpcore Loading…
1
Refine the gradient accumulation API
#9078 opened May 1, 2025 by rpsilva-aws • Draft
[cherry-pick to r2.6_aws_neuron] add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9077 opened May 1, 2025 by aws-yyjau Loading…
Revise the clone instruction in CONTRIBUTING.md for creating PRs
#9076 opened May 1, 2025 by junjieqian Loading…
1
1
Fix + Run DynamicShapeDetector tests on CI.
#9075 opened May 1, 2025 by ysiraichi Loading…
Disable running slow CI for doc-only changes
#9072 opened May 1, 2025 by ghpvnist Loading…
1
7
Replace xm.mark_step with torch_xla.sync() wherever possible
#9070 opened May 1, 2025 by ghpvnist Loading…
1
4
Remove trailing whitespace from the repo
#9065 opened Apr 30, 2025 by ghpvnist Loading…
1
Assert on empty PJRT buffers
#9062 opened Apr 29, 2025 by rpsilva-aws Loading…
1
add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9057 opened Apr 29, 2025 by aws-yyjau Loading…
2
Add clang link to PATH.
#9053 opened Apr 29, 2025 by ysiraichi Loading…
6
(Draft - PT27) Extend device data node binding to not override tensor IDs
#9052 opened Apr 28, 2025 by rpsilva-aws • Draft
Fix tensor factory when passing XLA device.
#9048 opened Apr 28, 2025 by ysiraichi • Draft
1
1
Fixes pytorch/xla#7398
#9047 opened Apr 26, 2025 by qianminj123 Loading…
2
reduce redundant graph in collective op distributed SPMD and other distributed things.
#9044 opened Apr 25, 2025 by zpcore Loading…
1
[DRAFT] call_jax sets the JAX contextual mesh
#9043 opened Apr 25, 2025 by tengyifei • Draft
Update test_export_fx_passes.py
#8933 opened Apr 3, 2025 by avikchaudhuri Loading…
1
add CreateGlobalShardedData prototype
#8932 opened Apr 3, 2025 by pgmoka • Draft
1
@pgmoka
test on debian12 DO_NOT_MERGE Not for merging.
#8928 opened Apr 2, 2025 by zpcore • Draft
8
Make GPU CUDA plugin require JAX
#8919 opened Apr 1, 2025 by tengyifei • Draft
Lower upsample_bilinear2d_backward operation.
#8916 opened Apr 1, 2025 by ysiraichi • Draft
1
1
[Draft] Add Experimental limited sparse embedding bag
#8905 opened Mar 28, 2025 by amjames • Draft
1
@lsy323
1
[ragged-paged-attn] Use hidden states in kv cache and support any num…
#8891 opened Mar 26, 2025 by zpcore Loading…
[DRAFT/WIP] Add top-p masking
#8871 opened Mar 21, 2025 by hyeygit • Draft
[1/N] Initial implementation of local SPMD support distributed SPMD and other distributed things.
#8810 opened Mar 9, 2025 by lsy323 Loading…
@lsy323
39
ProTip! Add no:assignee to see everything that’s not assigned.