pytorch/xla

Pull Requests Commits

support nested composite

chunnienc committed 1 year ago

7c922cb1

[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

Don't bundle libtpu in nightly (#6885)

alanwaketan committed 1 year ago

Verified a8f448d1

[Pallas] Make FlashAttention as torch.autograd.Function (#6886)

alanwaketan committed 1 year ago

Verified 0c704cf8

[benchmarks] Use Torchbench YAML configuration data. (#6881)

ysiraichi committed 1 year ago

Verified 7520c046

Fix vector norm lowering (#6883)

lsy323 committed 1 year ago

Verified f4371b67

Add flash attention as custom op so dynamo can trace it (#6875)

JackCaoG committed 1 year ago

Verified d9386809

Check pointer when we cast PjRtLayout to PjRtXlaLayout (#6882)

vanbasten23 committed 1 year ago

Verified 17823c3b

Fix div lowering and core aten test script enhancement (#6873)

lsy323 committed 1 year ago

Verified 895b0c2e

[Pallas] Support Flash Attention backward kernels (#6870)

alanwaketan committed 1 year ago

Verified c54367c8

[while_loop] prepare for torch while_loop signature change. (#6872)

ydwu4 committed 1 year ago

Verified 02f2a0dd

[OpenXLA|Moving TSL to XLA] Remove unused deps from TSL in CPP tests (#6868)

ManfeiBai committed 1 year ago

Verified 8cf394f2

update inst, remove some prints (#6866)

qihqi committed 1 year ago

Verified 88f93fe7

Add a `DynamoSyncInputExecuteTime` counter (#6813)

JackCaoG committed 1 year ago

Verified c1773b66

Refactor `nms` into TorchVision variant. (#6814)

ysiraichi committed 1 year ago

Verified 6cf9b915

Reland Install torchvision in release docker #6795 (#6860)

lsy323 committed 1 year ago

Verified b0ba29f9

Run pre-submit, post-submit TPU CI on the relesae branch (#6856)

lsy323 committed 1 year ago

Verified 23aa4ef8

Re-enable AMP test on GPU CI. (#6790)

vanbasten23 committed 1 year ago

Verified 5f858e7b

Older