Skip to content

Pull requests: pytorch/xla

New pull request New
138 Open 5,851 Closed
138 Open 5,851 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

Add impl for torchvision.nms and support multiple batch dim for convolution.
#8291 opened Oct 19, 2024 by qihqi Loading…
1
Fix op info test for svd
#8289 opened Oct 19, 2024 by dvhg Loading…
[WIP] enable tensorsolve
#8287 opened Oct 18, 2024 by barney-s Loading…
1
[DO NOT REVIEW] Experiment the pl.debug_print in Pallas.
#8284 opened Oct 18, 2024 by vanbasten23 • Draft
Update XLA pin to 10/16 tpuci
#8267 opened Oct 16, 2024 by JackCaoG • Draft
6
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…
1
[DO NOT REVIEW YET] Extend paged attention
#8237 opened Oct 8, 2024 by vanbasten23 • Draft
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…
14
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
ProTip! Follow long discussions with comments:>50.