Skip to content

Pull requests: pytorch/xla

New pull request New
149 Open 5,370 Closed
149 Open 5,370 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

Add helper function to get 1d mesh
#7577 opened Jun 26, 2024 by JackCaoG Loading…
1
[SPMD] Support manual all-reduce backport_2.4
#7576 opened Jun 26, 2024 by alanwaketan Loading…
@alanwaketan
6
move dynamo_mark_sharding and dynamo_set_buffer_donor_ out of experim…
#7502 opened Jun 25, 2024 by JackCaoG Loading…
1
Fix spmd reduce scatter python test
#7500 opened Jun 25, 2024 by bhavya01 Loading…
@bhavya01
Enable eager spmd eager SPMD
#7341 opened Jun 24, 2024 by JackCaoG • Draft
tests for tpu-info CLI tpuci usability Bugs/features related to improving the usability of PyTorch/XLA
#7337 opened Jun 24, 2024 by will-cromar • Draft
Add torch_pin to 2.4 release branch
#7336 opened Jun 24, 2024 by bhavya01 Loading…
@bhavya01
Update OpenXLA-pin to Jun24 2024Q2-manfei-review
#7334 opened Jun 24, 2024 by ManfeiBai • Draft
Update contributing doc after walkthrough
#7332 opened Jun 24, 2024 by tengyifei Loading…
9
Fix in-place ops for dynamo openxla_eval backend
#7320 opened Jun 20, 2024 by wonjoolee95 • Draft
Introduce CUDA OpenXLA fallback. xla:gpu
#7318 opened Jun 20, 2024 by ysiraichi Loading…
31
Update HLO instructions as per issue #7188 docathon-h1-2024 medium
#7317 opened Jun 19, 2024 by Arshi81099 Loading…
2
[torch_xla2] Prototype of torch.distributed backend
#7311 opened Jun 18, 2024 by will-cromar Loading…
3
[fori_loop|while_loop] add test case for Linear model with changeable weight
#7302 opened Jun 17, 2024 by ManfeiBai Loading…
@ManfeiBai
6
Use jax.export insteadl of jax.experimental.export
#7300 opened Jun 17, 2024 by qihqi • Draft
[1/N] Use std::string_view
#7297 opened Jun 16, 2024 by cyyever • Draft
2
Hanq bench
#7286 opened Jun 14, 2024 by qihqi • Draft
Modularize zero step function
#7265 opened Jun 13, 2024 by karakusc Loading…
[ZeRO-1] Only warn on rank-0
#7252 opened Jun 12, 2024 by hgt312 Loading…
Lower upsample_nearest_3d and its backward
#7251 opened Jun 12, 2024 by wonjoolee95 • Draft
Sample training UX
#7220 opened Jun 7, 2024 by qihqi • Draft
2
Add a test to convert xla gpu tensor to cuda tensor by using the protocol.
#7213 opened Jun 6, 2024 by vanbasten23 Loading…
[torch_xla2] Add lowerings for rand_like and randn_like, and enable some opinfo tests
#7170 opened Jun 3, 2024 by will-cromar Loading…
1
Add jlibrary module to torch_xla2 for preserving library abstractions during export
#7128 opened May 28, 2024 by GleasonK Loading…
@qihqi @lsy323
[OpenXLA-pin-update trigger] Create _openxla_pin_update_weekly.yml
#7108 opened May 24, 2024 by ManfeiBai Loading…
26
ProTip! Exclude everything labeled bug with -label:bug.