pytorch/xla

Pull Requests Commits

Add a devcontainer for gpu dev

JackCaoG committed 1 year ago

34fe27ba

Add dynamo expand test. (#6659)

ysiraichi committed 1 year ago

Verified 58a412cb

Build plugin during nightly build (#6917)

will-cromar committed 1 year ago

Verified 5a269d7b

Install libtpu for release dockers (#6919)

alanwaketan committed 1 year ago

Verified ec5615c9

[SPMD] Support manual sharding (#6915)

alanwaketan committed 1 year ago

Verified e5513ff4

fix pixel_shuffle return empty (#6907)

zpcore committed 1 year ago

Verified 43f10715

Build TPU package in CI (#6918)

will-cromar committed 1 year ago

Verified 7375a8ad

Add an option to not generate weights. (#6909)

qihqi committed 1 year ago

Verified fc11bbbf

Add `xla.sync` (#6914)

will-cromar committed 1 year ago

Verified 42452717

[Fori_loop|While_loop] Fori loop wrapped from while_loop (#6850)

ManfeiBai committed 1 year ago

Verified f0354ecc

[torch_xla2] Combined tensor.XLADispatchMode() and functions.XLAFunctionMode() (#6902)

will-cromar committed 1 year ago

Verified cfc70e68

Fix runtime error when run dynamo with a profiler scope (#6913)

JackCaoG committed 1 year ago

Verified 7dfbef02

Add ansible stage for building CUDA plugin (#6910)

will-cromar committed 1 year ago

Verified a170ffe6

[Distributed] Switch all_reduce to use the new functional collective op (#6887)

yifuwang committed 1 year ago

Verified a816c42f

fix wheel link (#6908)

zpcore committed 1 year ago

Verified 756b0eca

Update XLA pin to 04/08/2024 (#6904)

JackCaoG committed 1 year ago

Verified fd04a20c

[torch_xla2] Implement some common tensor constructors (#6874)

will-cromar committed 1 year ago

Verified 786484d2

[Pallas] Remove torch.empty in tracing (#6897)

alanwaketan committed 1 year ago

Verified 66ed39ba

Set all_reduce_token to null when exiting (#6898)

yitongh committed 1 year ago

Verified bd2b8fc8

update rc12 for 2.3 release (#6903)

lsy323 committed 1 year ago

Verified 1fe6be30

Add dynamo `torch.Tensor.new` test. (#6661)

ysiraichi committed 1 year ago

Verified 139d891e

Fix README and add partner links (#6896)

JackCaoG committed 1 year ago

Verified 5c48be19

add torch_distributed tests to our TPU CI (#6891)

JackCaoG committed 1 year ago

Verified 44bf47c4

Bump JAX version to match the latest pin update (#6893)

alanwaketan committed 1 year ago

Verified 77be46bd

[SPMD] UseVirtualDevice() is set, when XLA_AUTO_SPMD=1 (#6894)

yeounoh committed 1 year ago

Verified f44875c0

Manually init absl log to avoid log spam (#6890)

JackCaoG committed 1 year ago

Verified 62f6e05c

use init_method=xla:// for torch.distributed test (#6888)

JackCaoG committed 1 year ago

Verified 194ac22a

Update xla pin to Apr 3 (#6879)

bhavya01 committed 1 year ago

Verified 254970e5

Add opinfo test (#6878)

qihqi committed 1 year ago

Verified 0b9513cb

Adjust allclose atol for the flash attention TPU test (#6889)

JackCaoG committed 1 year ago

Verified 73de972b

Older