Skip to content

Pull requests: pytorch/xla

New pull request New
155 Open 5,175 Closed
155 Open 5,175 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

[Pallas] Make gmm functional tpuci
#7117 opened May 25, 2024 by alanwaketan Loading…
@alanwaketan
1
Add function for retrieving fallback operations. xla:gpu
#7116 opened May 24, 2024 by ysiraichi Loading…
index: fix index of 0-element tensor by 0-element tensor. xla:gpu
#7113 opened May 24, 2024 by ysiraichi Loading…
Only use remote cache for main repository, not forks
#7112 opened May 24, 2024 by will-cromar • Draft
upsample_bilinear: fix output data-type. xla:gpu
#7111 opened May 24, 2024 by ysiraichi Loading…
1
[OpenXLA-pin-update trigger] Create _openxla_pin_update_weekly.yml
#7108 opened May 24, 2024 by ManfeiBai Loading…
15
Disable kokoro and without key
#7106 opened May 23, 2024 by ManfeiBai Loading…
Enable while_loop/fori_loop, Add linear/MNIST and _get_xla_computation
#7094 opened May 22, 2024 by ManfeiBai • Draft
1
fix: configure Hermetic Python
#7092 opened May 22, 2024 by vam-google Loading…
13
update pin
#7087 opened May 21, 2024 by lsy323 • Draft
Add CI workflow for tests that requires pytorch CUDA.
#7073 opened May 16, 2024 by vanbasten23 Loading…
8
docs: fix typos in spmd.md
#7065 opened May 15, 2024 by clumsy Loading…
[test] Fori loop simple case test post order linear model wrapped body clean version while loop only new args fori_loop
#7064 opened May 15, 2024 by ManfeiBai • Draft
1
[test] Fori loop simple case test post order linear model wrapped body clean version while loop only
#7047 opened May 10, 2024 by ManfeiBai • Draft
Run persistent cache test on TPU CI tpuci
#7038 opened May 8, 2024 by jonb377 Loading…
3
Megablox tgmm kernel implementation pallas
#7032 opened May 6, 2024 by miladm Loading…
@miladm
2
[test] Fori loop simple case without hard-code
#7031 opened May 6, 2024 by ManfeiBai • Draft
4
Bump jinja2 from 3.1.3 to 3.1.4 in /docs dependencies Pull requests that update a dependency file
#7027 opened May 6, 2024 by dependabot bot Loading…
[Doc] Update spmd.md for doc documentation
#7019 opened May 2, 2024 by ManfeiBai Loading…
2
[WIP] Implement randn and empty
#7018 opened May 2, 2024 by will-cromar • Draft
[ZeRO-1] Sync features from r2.1_aws_neuron branch
#7016 opened May 1, 2024 by hgt312 Loading…
1
[Fori_loop|While_loop] Add MNIST test case with _xla_while_loop
#7012 opened May 1, 2024 by ManfeiBai • Draft
1
add check for neuron root comm id if not set
#7007 opened Apr 30, 2024 by aws-kingrj Loading…
3
Support ahead of time (AOT) cross-device compilation AOT ahead of time cross device compilation
#6992 opened Apr 29, 2024 by zpcore Loading…
4
[DO NOT REVIEW YET] Zero copy tensors between CUDA and XLA
#6970 opened Apr 25, 2024 by vanbasten23 • Draft
ProTip! Add no:assignee to see everything that’s not assigned.