Skip to content

Issues: pytorch/xla

[RFC] PyTorch/XLA Auto-Sharding API
#6322 opened Jan 18, 2024 by yeounoh
Open 10
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

399 Open 1,385 Closed
399 Open 1,385 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
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Distributed Checkpointing Saving Empty Files with 2D SPMD
#7118 opened May 25, 2024 by huzama
DDP Hangs on TPU v3-8
#7109 opened May 24, 2024 by vivekjoshy
6
Why does my 3-layer linear graph need to output two Transposes?
#7103 opened May 23, 2024 by mars1248
1
Problem with mesh shape in HybridMesh on TPU
#7102 opened May 23, 2024 by manh3152924
5
TPU Freezing on loss.backward() on same epoch.
#7101 opened May 23, 2024 by axhero7
1
[Feature] add cp311, cp312 support for XLA device
#7100 opened May 23, 2024 by Mon-ius
6
upsample_bilinear2d HLO returns unexpected data-type. xla:gpu
#7095 opened May 22, 2024 by ysiraichi
1
6
[torchbench] timm_nfnet training failing on non-dynamo. xla:gpu
#7084 opened May 20, 2024 by ysiraichi
1
1
Mismatch between XLA Tensor and PyTorch Native Tensor Results for torch.matmul in FP16 Precision on NVIDIA GPU
#7077 opened May 17, 2024 by lausannel
5
Cannot Import _XLAC
#7070 opened May 16, 2024 by DarkenStar
4
Experiencing slow recompilation when manually building XLA
#7057 opened May 14, 2024 by wfckl789
8
Export nn.Module.forward with kwargs to StableHLO stablehlo StableHLO related work
#7056 opened May 13, 2024 by johnmatter
2
torchdynamo + XLA crash
#7053 opened May 13, 2024 by pritamdamania87
10
multi_tensor_sgd triggers extra xla execution
#7051 opened May 13, 2024 by shenh10
2
The behavior of torch.einsum significantly differs between TPU and other devices.
#7050 opened May 13, 2024 by jqhoogland
1
Spmd whether expert parallelism is supported？
#7049 opened May 13, 2024 by mars1248
3
model.to(xla_device) increases the number of named_parameters
#7042 opened May 9, 2024 by shenh10
4
[torchbench] The official benchmark for performance and accuracy check
#7040 opened May 9, 2024 by shenh10
7
Migrate PyTorch/XLA's gradient checkpointing to upstream one nostale Do not consider for staleness
#7024 opened May 3, 2024 by JackCaoG
3
GPT2 CasualLM Inference crashes when using transformers v4.39.0
#6991 opened Apr 29, 2024 by sechkova
1
5
Using StableHLOCompositeBuilder in nested way.
#6978 opened Apr 26, 2024 by tlsdmstn56
3
Zero-copy between CUDA and XLA
#6971 opened Apr 25, 2024 by vanbasten23
5
Encountering out-of-memory errors despite using modest model and batch sizes.
#6948 opened Apr 20, 2024 by seanswyi
2
[RFC] Enable fori_loop(while_loop) in PyTorch/XLA fori_loop
#6941 opened Apr 18, 2024 by ManfeiBai
@ManfeiBai
op lowering for Upsample3D (aten::upsample_nearest3d)
#6932 opened Apr 17, 2024 by nzabari
@wonjoolee95
3
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.