pytorch/xla

Pull Requests Commits

Print the training update after the device execution

JackCaoG committed 2 years ago

c3d4f226

[torch_xla2] Basic DDP example in a notebook (#8042)

will-cromar committed 2 years ago

Verified ea8c47a3

[torch_xla2] Fix nn.functional.conv2d and conv3d (#8048)

dvhg committed 2 years ago

Verified 6af550aa

[torch_xla2] Fix scatter and scatter_add (#8045)

dvhg committed 2 years ago

Verified 2aea97a7

parallel_loader: Fix TPU memory leak when calling __iter__. (#8039)

dudulightricks committed 2 years ago

Verified d79a37c4

Update and short circuit the Neuron initialization (#8041)

rpsilva-aws committed 2 years ago

Verified d0ea5cc2

Op info test for linalg.norm, linalg.pinv, linalg.solve (#7503) (#8044)

yenkwang committed 2 years ago

Verified 441098b7

Fix minor grammar error (#8043)

simonteozw committed 2 years ago

Verified c0501f01

Fixed masked_scatter and masked_select (#8037)

anishfish2 committed 2 years ago

Verified 52673541

Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378) (#8038)

simonteozw committed 2 years ago

Verified b79a46b0

Fix quantile, nanquantile and gather (#8040)

matinehAkhlaghinia committed 2 years ago

Verified 6b33f8f6

Uncomment tests that are passing. (#8035)

cornmander committed 2 years ago

Verified 0415cc60

Fix unbind op. (#8033)

cornmander committed 2 years ago

Verified 4fe55f55

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

Older