pytorch/xla

Pull Requests Commits

this will crash after the epoch end and I am not sure why...

JackCaoG committed 2 years ago

669e59f4

Update SampleGenerator

JackCaoG committed 2 years ago

3539e6e6

Update comments in data parallel example to use sampler

JackCaoG committed 2 years ago

87a50b15

Implement empty_like in torch aten ops (#8029)

nupurbaghel committed 2 years ago

Verified 0da723fc

Cache flash attention tracing (#8026)

JackCaoG committed 2 years ago

Verified 8b936115

Remove contiguity assertion from `XLATensorImpl`. (#7998)

ysiraichi committed 2 years ago

Verified 03374cd0

Fix op test for `nn.functional.scaled_dot_product_attention"nn.functi… (#8025)

guyao committed 2 years ago

Verified 3b0097d2

add `masked.amin` .. `masked.mean` (#8014)

ManfeiBai committed 2 years ago

Verified 0e477391

Update nightly version in README.md (#8024)

bhavya01 committed 2 years ago

Verified 5ec51397

[torchxla2] Op Info test for fft (#8018)

qihqi committed 2 years ago

Verified 3f4b31bb

Add fixes for narrow_copy and narrow ops (#8021)

matinehAkhlaghinia committed 2 years ago

Verified 008004d9

Implement dist in torch aten ops (#8019)

nupurbaghel committed 2 years ago

Verified 1b1eea21

Implement operations `cummax`, `cummin` (#8013)

guyao committed 2 years ago

Verified 2a5897d4

Update Openxla-pin to Sep13, libtpu-pin to Sep13, jax to 0.4.33_nightly (#7916)

ManfeiBai committed 2 years ago

Verified 37894b08

Implement bucketize in torch aten ops (#7396) (#8007)

simonteozw committed 2 years ago

Verified dbdebfd9

Set xla_tpu_enable_flash_attention=false to enable libtpu pin update (#8008)

bhavya01 committed 2 years ago

Verified 9c7f0835

[benchmarks] Disallow `XRT` for XLA:CUDA. (#8006)

ysiraichi committed 2 years ago

Verified afbd7bd5

Add nn.functional.adaptive_avg_pool2d (#8010)

tracych477 committed 2 years ago

Verified abb07e41

OpInfo test for take -- trace (#8009)

qihqi committed 2 years ago

Verified 183d1d41

add `histogram` .. `linalg.eigvals` (#8002)

ManfeiBai committed 2 years ago

Verified 24bc4b2a

Op info test for logdet, linalg.slogdet (#7503) (#7958) (#8004)

yenkwang committed 2 years ago

Verified 3a9f53f4

Follow 2 releases grace period for API deprecation (#8003)

zpcore committed 2 years ago

Verified 99924801

Use a different str for device instead of meta (#7995)

qihqi committed 2 years ago

Verified 1b57d1e8

Update test_xla_sharding.py to take into account the number of devices (#8001)

bhavya01 committed 2 years ago

Verified fc03aef3

Fixed Masked.normalize and masked.prod (#7999)

anishfish2 committed 2 years ago

Verified 0535c889

add `heaviside ` .. `bfloat16` (#7994)

ManfeiBai committed 2 years ago

Verified 2097a6bb

Add the scripts used for lightning docs (#7991)

qihqi committed 2 years ago

Verified e2488e7f

[torch_xla2] Op coverage - torch.diag (#7961)

guyao committed 2 years ago

Verified 835a6a1f

[torch_xla2] Use openwebtext for DDP example (#7959)

will-cromar committed 2 years ago

Verified e6f1bde2

Add on_ready_event (#7984)

JackCaoG committed 2 years ago

Verified 8888217e

Older