issues
search
pytorch
/
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k
stars
467
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Introduce multi-node SPMD initialization for Neuron
#8046
rpsilva-aws
opened
1 week ago
0
[torch_xla2] Fix scatter and scatter_add
#8045
dvhg
closed
6 days ago
1
Op info test for linalg.norm, linalg.pinv, linalg.solve (#7503)
#8044
yenkwang
closed
1 week ago
0
Fix minor grammar error
#8043
simonteozw
closed
1 week ago
0
[torch_xla2] Basic DDP example in a notebook
#8042
will-cromar
closed
6 days ago
0
Update and short circuit the Neuron initialization
#8041
rpsilva-aws
closed
1 week ago
0
Fix quantile, nanquantile and gather
#8040
matinehAkhlaghinia
closed
1 week ago
0
parallel_loader: Fix TPU memory leak when calling __iter__.
#8039
dudulightricks
closed
1 week ago
4
Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378)
#8038
simonteozw
closed
1 week ago
1
Fixed masked_scatter and masked_select
#8037
anishfish2
closed
1 week ago
0
add `resize_`, `resize_as_`, `rot90`, `repeat_interleave`, `combinations`, `diag_embed`
#8036
ManfeiBai
closed
5 days ago
0
Uncomment tests that are passing.
#8035
cornmander
closed
1 week ago
0
Add profiler annotation for the decoderonly example
#8034
JackCaoG
opened
1 week ago
0
Fix unbind op.
#8033
cornmander
closed
1 week ago
1
Forward `XLATensorImpl::is_contiguous_custom` to `TensorImpl`.
#8032
ysiraichi
opened
1 week ago
8
Add support for native_layer_norm
#8031
matinehAkhlaghinia
opened
1 week ago
1
[torchxla2] filter passed ops
#8030
ManfeiBai
closed
1 week ago
1
Implement empty_like in torch aten ops
#8029
nupurbaghel
closed
1 week ago
0
Implement empty_like in torch aten ops
#8028
nupurbaghel
closed
1 week ago
0
Add stablediffusion inference reference model
#8027
qihqi
closed
3 days ago
0
Cache flash attention tracing
#8026
JackCaoG
closed
1 week ago
0
Fix op test for `nn.functional.scaled_dot_product_attention"nn.functi…
#8025
guyao
closed
1 week ago
0
Update nightly version in README.md
#8024
bhavya01
closed
1 week ago
0
[benchmarks] Add docs file.
#8023
ysiraichi
opened
1 week ago
1
Add documentation for `pip install[pallas]`
#8022
miladm
opened
1 week ago
0
Add fixes for narrow_copy and narrow ops
#8021
matinehAkhlaghinia
closed
1 week ago
0
Distributed training on multi-host v4/v5 TPU Pods is too slow
#8020
ayukh
opened
1 week ago
6
Implement dist in torch aten ops
#8019
nupurbaghel
closed
1 week ago
0
[torchxla2] Op Info test for fft
#8018
qihqi
closed
1 week ago
0
[hard] Op info test for `masked.median`
#8017
ManfeiBai
opened
2 weeks ago
1
[hard] add `masked.median`
#8016
ManfeiBai
closed
2 weeks ago
1
[torch_xla2] Fix aten.cat lowering
#8015
chunnienc
closed
2 weeks ago
0
add `masked.amin` .. `masked.mean`
#8014
ManfeiBai
closed
1 week ago
0
Implement operations `cummax`, `cummin`
#8013
guyao
closed
2 weeks ago
0
xm.send_cpu_data_to_device cannot support 2d data and 4d mesh
#8012
fengyang0317
opened
2 weeks ago
1
Limit `XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shapes.py -v` only run on v4
#8011
ManfeiBai
closed
2 weeks ago
1
Add nn.functional.adaptive_avg_pool2d
#8010
tracych477
closed
2 weeks ago
0
OpInfo test for take -- trace
#8009
qihqi
closed
2 weeks ago
0
Set xla_tpu_enable_flash_attention=false to enable libtpu pin update
#8008
bhavya01
closed
2 weeks ago
0
Implement bucketize in torch aten ops (#7396)
#8007
simonteozw
closed
2 weeks ago
0
[benchmarks] Disallow `XRT` for XLA:CUDA.
#8006
ysiraichi
closed
2 weeks ago
0
[fp8] Support `fp8e4m3` in torch_xla
#8005
miladm
opened
2 weeks ago
1
Op info test for logdet, linalg.slogdet (#7503) (#7958)
#8004
yenkwang
closed
2 weeks ago
0
Follow 2 releases grace period for API deprecation
#8003
zpcore
closed
2 weeks ago
0
add `histogram` .. `linalg.eigvals`
#8002
ManfeiBai
closed
2 weeks ago
0
Update test_xla_sharding.py to take into account the number of devices
#8001
bhavya01
closed
2 weeks ago
0
[RFC] `torch_xla` Backward Compatibility Proposal
#8000
zpcore
opened
2 weeks ago
2
Fixed Masked.normalize and masked.prod
#7999
anishfish2
closed
2 weeks ago
0
Remove contiguity assertion from `XLATensorImpl`.
#7998
ysiraichi
closed
1 week ago
2
[hard] Op info test for `repeat_interleave`
#7997
ManfeiBai
opened
2 weeks ago
1
Previous
Next