pytorch/xla

Pull Requests Commits

remove safe_zip

yaochengji committed 1 year ago

6ea16540

remove safe_zip

yaochengji committed 1 year ago

5f3fa3cd

address comments

yaochengji committed 1 year ago

0bbf02d3

address comments

yaochengji committed 1 year ago

7a15ecfc

address comments

yaochengji committed 1 year ago

50334a63

yaochengji committed 1 year ago

715a3929

[Bugfix] fix ragged attention kernel auto-tuning table key

yaochengji committed 1 year ago

3b99bc8f

Fix python 3.11 cuda wheel link in the readme (#9493)

vfdev-5 committed 1 year ago

Verified e82631ec

Misc fixes: (#9491)

qihqi committed 1 year ago

Verified 26def0f2

Fix nested stableHLO composite regions (#9385)

Carlomus committed 1 year ago

Verified 55b7d021

Allow mixed tensor type math if one of them is a scalar (#9453)

qihqi committed 1 year ago

Verified 95ba754a

Remove support for one-process-per-device style of distributed. (#9490)

qihqi committed 1 year ago

Verified f3c7907f

Update README.md to reflect supported python versions (#9484)

bhavya01 committed 1 year ago

Verified 86a99d75

Add support for callable in torchax.interop.JittableModule.functional_call in the first parameter (#9451)

zmelumian972 committed 1 year ago

Verified ca914452

Implement collective gather op (#9435)

bfolie committed 1 year ago

Verified a021bf09

implement collective all_to_all op (#9442)

bfolie committed 1 year ago

Verified e7dcc7bd

Rename `XLA_SHOW_CPP_ERROR_CONTEXT` to `TORCH_SHOW_CPP_STACKTRACES` (#9482)

ysiraichi committed 1 year ago

Verified 58592c48

Improve error message for shape promotion on lowering. (#9486)

ysiraichi committed 1 year ago

Verified 2c556e73

Add r2.9 release nightly tests to README (#9478)

pgmoka committed 1 year ago

Verified 8b95c5d6

Fix spmd sharding visualization when device index is >= 10 (#9475)

jeffhataws committed 1 year ago

Verified c6aa90f1

Update Setuptools version in build dependencies. (#9487)

bhavya01 committed 1 year ago

Verified b612613d

Fix duplicate labels and other docs build warnings (#9446)

melissawm committed 1 year ago

Verified d7c9e506

Bump version (#9480)

qihqi committed 1 year ago

Verified 4b3d34da

Error Handling: refactor the existing `ComputationClient` implementations to use status QOL functions. (#9420)

ysiraichi committed 1 year ago

Verified 866fd22d

Optimize w8a8 pallas kernel (#9473)

kyuyeunk committed 1 year ago

Verified 36ff641d

Torchax: Allow reuse of the jittable procedure with different functio… (#9374)

zmelumian972 committed 1 year ago

Verified ebf9a8be

Update TPU CI with latest docker container (#9483)

bhavya01 committed 1 year ago

Verified ca04583b

Add dtensor mesh conversion test (#9474)

aws-cph committed 1 year ago

Verified 93a5e583

implement collective reduce op (#9437)

bfolie committed 1 year ago

Verified 173a7708

Error Handling: replace `ConsumeValue` with `GetValueOrThrow`. (#9464)

ysiraichi committed 1 year ago

Verified 51518f9b

Older