Skip to content

Pull requests: pytorch/xla

New pull request New
145 Open 5,114 Closed
145 Open 5,114 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

[WIP] Add dlpack support
DO_NOT_REVIEW_YET
#7025 opened May 4, 2024 by vanbasten23 • Draft
[Doc] Update spmd.md for doc
documentation
#7019 opened May 2, 2024 by ManfeiBai Loading…
[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
Support torch_xla2 benchmarking using torchbench
#7013 opened May 1, 2024 by zpcore Loading…
20
[Fori_loop|While_loop] Add MNIST test case with _xla_while_loop
#7012 opened May 1, 2024 by ManfeiBai • Draft
Move op dispatching logic into an Environment class; and use Mode to capture dispatcher instead of tensor.
#7009 opened Apr 30, 2024 by qihqi Loading…
4
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…
1
[DO NOT REVIEW YET] Zero copy tensors between CUDA and XLA
#6970 opened Apr 25, 2024 by vanbasten23 • Draft
Adding megablox gmm standalone
pallas
#6940 opened Apr 18, 2024 by miladm Loading…
@miladm
2
[test] Fori loop simple case test test
#6938 opened Apr 17, 2024 by ManfeiBai • 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…
format the output model input
#6869 opened Apr 1, 2024 by zpcore Loading…
1
4
[Fori_loop|While_loop] Modify XlaComputation and add linear model test case
#6867 opened Apr 1, 2024 by ManfeiBai Loading…
13
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
6
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
add CUDA_NUM_DEVICES env support
#6780 opened Mar 20, 2024 by Mon-ius Loading…
2
Register reshape to RegisterAutogradXLA.cpp
backport_2.3
#6752 opened Mar 14, 2024 by wonjoolee95 Loading…
1
9
Prototype bucketing for dynamic shapes in Dynamo Bridge
DO_NOT_MERGE Not for merging.
#6728 opened Mar 12, 2024 by changm Loading…
@changm
[DO NOT REVIEW] Testing CI
#6716 opened Mar 11, 2024 by vanbasten23 Loading…
1
[DO NOT REVIEW] Check the set values in the old TPU CI
#6706 opened Mar 8, 2024 by vanbasten23 Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.