pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

dcb859c5

Address commetns

alanwaketan committed 1 year ago

2c71054b

alanwaketan committed 1 year ago

36027f0f

alanwaketan committed 1 year ago

3cc5271b

alanwaketan committed 1 year ago

f58c8e78

alanwaketan committed 1 year ago

02a15f0b

Add a testcase somewhat

alanwaketan committed 1 year ago

675c8b34

alanwaketan committed 1 year ago

39e1dead

alanwaketan committed 1 year ago

d18e1082

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

[Pallas] Support multiple outputs (#6844)

alanwaketan committed 1 year ago

Verified 8f095fc8

Support array attribute in stablehlo composite (#6840)

chunnienc committed 1 year ago

Verified 7bbe9d7a

[SPMD] use UNKNOWN as default sharded data sharding type only for auto-sharding (#6846)

yeounoh committed 1 year ago

Verified c1db8759

update 2.3 rc-6 in build triggers (#6845)

lsy323 committed 1 year ago

Verified cba3fe17

update nightly version number (#6842)

lsy323 committed 1 year ago

Verified f6d6286d

Add a dev image for cuda12.3 (#6835)

bhavya01 committed 1 year ago

Verified 59db0cd4

[Pallas] Set the major-minor layout for inputs and outputs (#6826)

alanwaketan committed 1 year ago

Verified d99ceb6b

remove using_pjrt in xla_graph_executor (#6768)

JackCaoG committed 1 year ago

Verified d6fb5391

Older