Skip to content

Pull requests: pytorch/xla

New pull request New
142 Open 5,229 Closed
142 Open 5,229 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

torchbench CI CI CI related change
#7162 opened May 31, 2024 by zpcore • Draft
[Fori_loop|While_loop] Enable while_loop/fori_loop, Add linear/MNIST test case #7094
#7157 opened May 30, 2024 by ManfeiBai Loading…
Test upstream change about the sawp tensor DO_NOT_MERGE Not for merging.
#7156 opened May 30, 2024 by JackCaoG Loading…
Use dest_offsets directly in LoadPlanner
#7155 opened May 30, 2024 by jonb377 Loading…
isolate the PJRT dependency between torch_xla and torch_xla2
#7144 opened May 29, 2024 by zpcore Loading…
9
[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
9
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
[OpenXLA-pin-update trigger] Create _openxla_pin_update_weekly.yml
#7108 opened May 24, 2024 by ManfeiBai Loading…
26
fix: configure Hermetic Python
#7092 opened May 22, 2024 by vam-google Loading…
13
docs: fix typos in spmd.md
#7065 opened May 15, 2024 by clumsy Loading…
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
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…
[torch_xla2] Use torch.Generator for random number generation in Environment
#7018 opened May 2, 2024 by will-cromar Loading…
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
Remove old TPU CI DO_NOT_MERGE_YET For PRs which cannot be merged, despite tests passing
#6926 opened Apr 16, 2024 by will-cromar Loading…
7
Update LICENSE
#6911 opened Apr 10, 2024 by mshudrak Loading…
[torch_xla2] Use ruff for formatting
#6906 opened Apr 9, 2024 by will-cromar Loading…
Using dummy store when user skips barrier to save open file descriptors
#6834 opened Mar 27, 2024 by barry-jin Loading…
5
Triton Triton
#6798 opened Mar 21, 2024 by bhavya01 Loading…
@bhavya01
37
Add native torch profiler to spmd resnet model. DO_NOT_MERGE Not for merging. DO_NOT_REVIEW_YET
#6784 opened Mar 20, 2024 by vanbasten23 • Draft
ProTip! Follow long discussions with comments:>50.