issues
search
pytorch
/
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k
stars
483
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Export training model to StableHlo
#8366
Zantares
opened
2 weeks ago
2
Add tengyifei to docs code owners
#8365
tengyifei
closed
2 weeks ago
0
Add mikegre-google as docs owner
#8364
tengyifei
closed
2 weeks ago
1
Align rrelu_with_noise schema to reflect noise mutation [branch]
#8363
IvanKobzarev
opened
2 weeks ago
0
WIP implement lowering for torchvision::roi_align
#8362
barney-s
opened
2 weeks ago
0
Expose lazy all number special scalars
#8361
rpsilva-aws
closed
1 week ago
1
Add aten::lift_fresh to the exception list that converts input to torch tensor
#8360
barney-s
closed
2 weeks ago
0
Query regarding using 1 chip (2 cores of TPU v3) for Inference
#8359
deepakkumar2440
opened
2 weeks ago
0
Add inplace ops for sigmoid and tanh
#8358
barney-s
closed
2 weeks ago
0
Correct the cxx11_abi condition
#8357
tengyifei
closed
3 weeks ago
2
Support the case where query_len>context_len in the multi-queries paged attention.
#8356
vanbasten23
closed
1 week ago
8
Offer user guide instructions to users to leverage various `libtpu` versions
#8355
miladm
opened
3 weeks ago
1
Make XLA tests export new IR compatible
#8354
tugsbayasgalan
closed
2 weeks ago
0
Two different graph have the same hash value
#8353
yitongh
opened
3 weeks ago
1
Remove unnecessary functions and print statements (#7495)
#8352
simonteozw
closed
3 weeks ago
1
Fix torch.linalg.matrix_power (#7495)
#8351
simonteozw
closed
3 weeks ago
0
Add functions to emit custom call to place a buffer to host and device.
#8350
qihqi
closed
3 weeks ago
1
XLA downcast for S64 (and U64) for Neuron
#8349
aws-nm9
closed
1 week ago
0
DLPack: fix test using PyTorch API.
#8348
ysiraichi
opened
3 weeks ago
1
XLA downcast for S64 and U64 for Neuron
#8347
aws-nm9
closed
2 weeks ago
1
RuntimeError: torch_xla/csrc/data_ops.cpp:185 : Check failed: input_sizes.size() <= output_sizes.size()
#8346
samuele-bortolato1
opened
3 weeks ago
0
Error when usecollective communication in torch_xla.core.xla_model.all_to_all in SPMD mopde
#8345
DarkenStar
opened
3 weeks ago
0
Remove torch_pin that is accidentally introduced here:
#8344
qihqi
closed
3 weeks ago
0
Updates Contributing.md
#8343
mikegre-google
opened
3 weeks ago
0
Instructions in CONTRIBUTING.md for using VS Code don't seem to work
#8342
mikegre-google
closed
3 weeks ago
1
Fixing missing sharding_scheme param
#8341
slabhs
closed
3 weeks ago
0
R2.5.1
#8340
ManfeiBai
closed
3 weeks ago
0
[2.5.1 release] Update artifacts.auto.tfvars
#8339
ManfeiBai
closed
3 weeks ago
0
Update CODEOWNERS
#8338
zpcore
closed
3 weeks ago
0
R2.5.1
#8337
ManfeiBai
closed
3 weeks ago
0
[2.5.1 release] Update artifacts.auto.tfvars for 2.5.1 release
#8336
ManfeiBai
closed
3 weeks ago
0
[Neuron] Fix the XLADevice Neuron mappings for SPMD downcasts
#8335
rpsilva-aws
closed
3 weeks ago
2
Bug - Using Sharding in Flash Attention with segment ids.
#8334
dudulightricks
opened
3 weeks ago
0
custom_kernel: fix shape mismatch by sharding segment_ids in flash attn.
#8333
dudulightricks
opened
3 weeks ago
3
Revive "Modularize zero step function and make it customizable" #7233
#8332
jeffhataws
closed
2 weeks ago
0
Expensive PyTorch to StableHLO conversion
#8331
Wheest
opened
3 weeks ago
0
Convert entire model to jax xla
#8330
barney-s
closed
3 weeks ago
0
Unify shallow_copy_and_detach
#8329
cyyever
opened
4 weeks ago
0
Extend paged attention to support query_len>1
#8328
vanbasten23
closed
3 weeks ago
0
Add documentations for persistent caching
#8327
miladm
opened
1 month ago
0
[release] add permission of iam.auto.tfvars
#8326
ManfeiBai
opened
1 month ago
2
Add CODEOWNERS for /infra
#8325
ManfeiBai
closed
1 month ago
0
[2.5.1 release] [Wait for wheel build] Update r2.5.1 branch readme
#8324
ManfeiBai
closed
3 weeks ago
0
R2.5.1
#8323
ManfeiBai
closed
1 month ago
0
Fixes issue #7774
#8322
mikegre-google
closed
3 weeks ago
1
Fixed link to troubleshooting.md
#8321
mikegre-google
closed
3 weeks ago
0
[2.5.1 release] Update artifacts.auto.tfvars for 2.5.1
#8320
ManfeiBai
closed
1 month ago
0
R2.5.1
#8319
ManfeiBai
closed
1 month ago
0
Update version to 2.5.1
#8318
ManfeiBai
closed
1 month ago
0
[cherry-pick] R2.5.1 cherry pick #8315
#8317
ManfeiBai
closed
1 month ago
0
Previous
Next