pytorch/xla

Pull Requests Commits

Merge branch 'master' of https://github.com/pytorch/xla into bfolie/reduce-collective

bfolie committed 153 days ago

c25ee416

Add dtensor placement test (#9458)

jeffhataws committed 154 days ago

Verified cc151112

Update torch compat version to 2.7.1 (#9455)

qihqi committed 154 days ago

Verified 01e579c0

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

vanbasten23 committed 154 days ago

Verified 83dc9da8

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

pgmoka committed 154 days ago

Verified 13403086

Update cuda version check (#9469)

pgmoka committed 154 days ago

Verified d450ab37

Fix status source code location logic. (#9440)

ysiraichi committed 154 days ago

Verified 6acc8f3f

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

ysiraichi committed 154 days ago

Verified 12e77e72

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

ysiraichi committed 154 days ago

Verified 5496a364

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

ysiraichi committed 154 days ago

Verified 1d848e76

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

zzzwen committed 155 days ago

Verified 1c926f72

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

vanbasten23 committed 155 days ago

Verified cf156c68

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

vanbasten23 committed 155 days ago

Verified 52569ecc

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

tengyifei committed 155 days ago

Verified 156d9134

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

qihqi committed 155 days ago

Verified 8999ba55

Change nightly_package_version to 2.9 (#9461)

pgmoka committed 155 days ago

Verified 1ccfede9

Partially disable tpu-info CLI tests (#9463)

bhavya01 committed 155 days ago

Verified 33e87915

Fix CPU tests for python 3.12 (#9443)

bhavya01 committed 155 days ago

Verified 5d05f0a3

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

bfolie committed 157 days ago

Verified 66ad14f9

remove reduce from list of unimplemented backend methods

bfolie committed 162 days ago

9f9f04a3

Update TPU CI container image (#9434)

bhavya01 committed 162 days ago

Verified 1c00dea2

Add JAX dependency for Python 3.12 (#9438)

tengyifei committed 162 days ago

Verified 53ea5862

implement reduce op

bfolie committed 163 days ago

804acd73

Error Handling: refactor `XlaCoordinator` to use status types. (#9386)

ysiraichi committed 163 days ago

Verified 5b8ddb12

Add Build Trigger for 2.8-rc1 release (#9425)

pgmoka committed 163 days ago

Verified 752ddba7

Update CI docker images to use 3.12 (#9427)

bhavya01 committed 163 days ago

Verified 773ecdfb

Optimize w8a8 quantized matmul kernel (#9412)

vanbasten23 committed 163 days ago

Verified 4101ea56

Install libtpu directly instead of torch_xla[tpu] (#9423)

bhavya01 committed 164 days ago

Verified 9c8ae9f9

`EmbeddingDenseBackward`: Remove `padding_idx` cast to `double` (#9406)

unterumarmung committed 165 days ago

Verified 3d92aa75

s/torch_xla2/torchax (#9353)

tengyifei committed 166 days ago

Verified 9499e6f4

Older