pytorch/xla

Pull Requests Commits

Update Setuptools version in build dependencies.

bhavya01 committed 1 year ago

Verified eb7a8066

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

[Kernel] Update ragged attention block table (#9476)

yaochengji committed 1 year ago

Verified 992e87ad

Add dtensor placement test (#9458)

jeffhataws committed 1 year ago

Verified cc151112

Update torch compat version to 2.7.1 (#9455)

qihqi committed 1 year ago

Verified 01e579c0

Calculate vmem limit dynamically in the quantized matmul kernel. (#9470)

vanbasten23 committed 1 year ago

Verified 83dc9da8

Remove duplicate artifact creation for 2.8.0-rc1 (#9471)

pgmoka committed 1 year ago

Verified 13403086

Update cuda version check (#9469)

pgmoka committed 1 year ago

Verified d450ab37

Fix status source code location logic. (#9440)

ysiraichi committed 1 year ago

Verified 6acc8f3f

Error Handling: replace `XLA_CHECK_OK()` with status functions. (#9457)

ysiraichi committed 1 year ago

Verified 12e77e72

Suppress C++ stacktrace on `XLA_CHECK*()` calls. (#9448)

ysiraichi committed 1 year ago

Verified 5496a364

Error Handling: refactor the PjRt registry to use status QOL functions. (#9419)

ysiraichi committed 1 year ago

Verified 1d848e76

By default, to("jax") should go to TPU (#9468)

zzzwen committed 1 year ago

Verified 1c926f72

Remove the clamp op when we do symmetric quantization on a tensor (#9465)

vanbasten23 committed 1 year ago

Verified cf156c68

Unify the return type of w8a8 matmul between fallback and the actual impl. (#9452)

vanbasten23 committed 1 year ago

Verified 52569ecc

Support editable install with setuptools>=80.0.0 (#9428)

tengyifei committed 1 year ago

Verified 156d9134

Make assume_pure able to work with functions that depends on random (#9460)

qihqi committed 1 year ago

Verified 8999ba55

Change nightly_package_version to 2.9 (#9461)

pgmoka committed 1 year ago

Verified 1ccfede9

Partially disable tpu-info CLI tests (#9463)

bhavya01 committed 1 year ago

Verified 33e87915

Fix CPU tests for python 3.12 (#9443)

bhavya01 committed 1 year ago

Verified 5d05f0a3

Change update_deps script so that latest stable version can be pulled instead of latest nightly (#9424)

bfolie committed 1 year ago

Verified 66ad14f9

Update TPU CI container image (#9434)

bhavya01 committed 1 year ago

Verified 1c00dea2

Older