pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

99dc7c0a

Support test__flash_attention_bwd_dkv

alanwaketan committed 1 year ago

e01dead5

Support test__flash_attention_bwd_dkv

alanwaketan committed 1 year ago

cf9a1d07

support test__flash_attention_impl

alanwaketan committed 1 year ago

1ce54b1a

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

Remove references to XRT in build and CI tooling (#6761)

will-cromar committed 1 year ago

Verified f083e105

Disable failed C++ Test (#6823)

lsy323 committed 1 year ago

Verified 2b66e93f

Add experimental MLIR debuginfo writer API (#6799)

chunnienc committed 1 year ago

Verified 64a146d8

Use fsspec for multihost checkpoint (#6818)

jonb377 committed 1 year ago

Verified 1ad6bb45

[Pallas] Support XLA_USE_BF16 (#6817)

alanwaketan committed 1 year ago

Verified 73c31dbe

Revert "Install torchvision in release docker" (#6821)

lsy323 committed 1 year ago

Verified b8a97dfc

Add missing shape metadata for the last replaced fx node (#6805)

lsy323 committed 1 year ago

Verified 899a0fa4

Revert "[Reland]Change ATEN generator argument type to const std::optional<Generator>&" (#6819)

alanwaketan committed 1 year ago

Verified 134f5b62

[Fori_loop|While_loop] Add nested test (#6807)

ManfeiBai committed 1 year ago

Verified 22fe1dc3

Install torchvision in release docker (#6795)

lsy323 committed 1 year ago

Verified 7b9177de

Remove reshape lowering (#6801)

wonjoo-wj committed 1 year ago

Verified a84509b2

[Reland]Change ATEN generator argument type to const std::optional<Generator>& (#6804)

cyyever committed 1 year ago

Verified 84e7feb3

Older