Skip to content

Pull requests: pytorch/xla

New pull request New
160 Open 5,189 Closed
160 Open 5,189 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

Facebook->Meta in README
#7141 opened May 29, 2024 by kit1980 Loading…
[DO NOT REVIEW YET]add workflow for test that requires pytorch CUDA.
#7140 opened May 29, 2024 by vanbasten23 • Draft
[Pallas] Support tgmm tpuci
#7137 opened May 29, 2024 by alanwaketan Loading…
@alanwaketan
Update contribute to include bdist_wheel
#7135 opened May 29, 2024 by JackCaoG Loading…
2
[ZeRO-1] Sync features from r2.1_aws_neuron branch
#7132 opened May 29, 2024 by hgt312 Loading…
mul: convert inputs to result type. xla:gpu
#7130 opened May 28, 2024 by ysiraichi Loading…
1
8
Add examples to show case llama training.
#7129 opened May 28, 2024 by qihqi Loading…
Add jlibrary module to torch_xla2 for preserving library abstractions during export
#7128 opened May 28, 2024 by GleasonK Loading…
@qihqi @lsy323
add test for onehot to make sure no fallback
#7127 opened May 28, 2024 by JackCaoG Loading…
1
Make from_dlpack handle cuda synchronization implicitly for input tensors that have __dlpack__ and __dlpack_device__ attributes.
#7125 opened May 28, 2024 by vanbasten23 Loading…
4
[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
ProTip! Add no:assignee to see everything that’s not assigned.