pytorch/xla

Pull Requests Commits

remove safe_zip

yaochengji committed 313 days ago

6ea16540

remove safe_zip

yaochengji committed 313 days ago

5f3fa3cd

address comments

yaochengji committed 314 days ago

0bbf02d3

address comments

yaochengji committed 314 days ago

7a15ecfc

address comments

yaochengji committed 314 days ago

50334a63

yaochengji committed 314 days ago

715a3929

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

yaochengji committed 314 days ago

3b99bc8f

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

vfdev-5 committed 315 days ago

Verified e82631ec

Misc fixes: (#9491)

qihqi committed 316 days ago

Verified 26def0f2

Fix nested stableHLO composite regions (#9385)

Carlomus committed 316 days ago

Verified 55b7d021

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

qihqi committed 318 days ago

Verified 95ba754a

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

qihqi committed 318 days ago

Verified f3c7907f

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

bhavya01 committed 318 days ago

Verified 86a99d75

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

zmelumian972 committed 318 days ago

Verified ca914452

Implement collective gather op (#9435)

bfolie committed 318 days ago

Verified a021bf09

implement collective all_to_all op (#9442)

bfolie committed 319 days ago

Verified e7dcc7bd

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

ysiraichi committed 319 days ago

Verified 58592c48

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

ysiraichi committed 319 days ago

Verified 2c556e73

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

pgmoka committed 319 days ago

Verified 8b95c5d6

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

jeffhataws committed 319 days ago

Verified c6aa90f1

Update Setuptools version in build dependencies. (#9487)

bhavya01 committed 320 days ago

Verified b612613d

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

melissawm committed 320 days ago

Verified d7c9e506

Bump version (#9480)

qihqi committed 320 days ago

Verified 4b3d34da

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

ysiraichi committed 320 days ago

Verified 866fd22d

Optimize w8a8 pallas kernel (#9473)

kyuyeunk committed 320 days ago

Verified 36ff641d

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

zmelumian972 committed 320 days ago

Verified ebf9a8be

Update TPU CI with latest docker container (#9483)

bhavya01 committed 320 days ago

Verified ca04583b

Add dtensor mesh conversion test (#9474)

aws-cph committed 321 days ago

Verified 93a5e583

implement collective reduce op (#9437)

bfolie committed 322 days ago

Verified 173a7708

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

ysiraichi committed 324 days ago

Verified 51518f9b

Older