Skip to content

Pull requests: pytorch/xla

New pull request New
136 Open 5,819 Closed
136 Open 5,819 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

Fix op info test for linalg.inv and linalg.inv_ex
#8253 opened Oct 11, 2024 by dvhg Loading…
update runtime error message for minibatch
#8243 opened Oct 9, 2024 by JackCaoG Loading…
1
Fix test_arange_nan after introduction of checks to out= functions.
#8242 opened Oct 9, 2024 by ysiraichi Loading…
Add _aten_cdist Op
#8238 opened Oct 9, 2024 by vyom1611 Loading…
1
[DO NOT REVIEW YET] Extend paged attention
#8237 opened Oct 8, 2024 by vanbasten23 • Draft
[test] update XLA commit to JAX 0.4.33 same one 2.5 release
#8229 opened Oct 7, 2024 by ManfeiBai Loading…
1
Pin update
#8079 opened Sep 26, 2024 by qihqi Loading…
Check warning msg in second captured warning.
#8069 opened Sep 25, 2024 by zhuhaozhe Loading…
Support FSDPv2 compute dtype
#8056 opened Sep 24, 2024 by lausannel Loading…
4
Print the training update after the device execution DO_NOT_REVIEW_YET
#8054 opened Sep 24, 2024 by JackCaoG • Draft
Part 2. Introduce multi-node SPMD initialization for Neuron
#8046 opened Sep 19, 2024 by rpsilva-aws • Draft
2
Forward XLATensorImpl::is_contiguous_custom to TensorImpl.
#8032 opened Sep 17, 2024 by ysiraichi Loading…
12
Add support for native_layer_norm
#8031 opened Sep 17, 2024 by matinehAkhlaghinia Loading…
1
[benchmarks] Add docs file.
#8023 opened Sep 16, 2024 by ysiraichi Loading…
7
support dist.broadcast tpuci usability Bugs/features related to improving the usability of PyTorch/XLA
#7956 opened Sep 5, 2024 by zpcore Loading…
6
Update dedup logic to use dedup_save_plans
#7929 opened Aug 29, 2024 by jonb377 Loading…
4
Update comments in data parallel example to use sampler
#7914 opened Aug 26, 2024 by JackCaoG Loading…
1
6
scan and apply_layers
#7901 opened Aug 23, 2024 by tengyifei Loading…
@tengyifei
56
[WIP] Skip AOTAutograd view-reconstruction for inference runs. xla:gpu
#7834 opened Aug 12, 2024 by ysiraichi • Draft
1
API_GUIDE.md: fix execution => exception typo + rewording
#7808 opened Aug 6, 2024 by houeland Loading…
Support SplashAttention in PyTorch/XLA tpuci
#7798 opened Aug 1, 2024 by wonjoolee95 Loading…
2
usability
#7728 opened Jul 23, 2024 by zpcore • Draft
Support of AOT compilation (refine #6992) AOT ahead of time cross device compilation
#7581 opened Jun 26, 2024 by zpcore Loading…
8
Update HLO instructions as per issue #7188 docathon-h1-2024 medium
#7317 opened Jun 19, 2024 by Arshi81099 Loading…
8
[fori_loop|while_loop] add test case for Linear model with changeable weight
#7302 opened Jun 17, 2024 by ManfeiBai Loading…
@ManfeiBai
19
ProTip! Exclude everything labeled bug with -label:bug.