pytorch/xla

Pull Requests Commits

Update artifacts_builds.tf for 2.8.0-rc4

bhavya01 committed 135 days ago

Verified afaeea8b

Introduce PlatformVersion bindings (#9513)

rpsilva-aws committed 135 days ago

Verified 0ad39c22

Support torch.nn.functional.one_hot (#9523)

vanbasten23 committed 135 days ago

Verified 7a48185d

Reorganize PyTorch/XLA Overview page (#9498)

melissawm committed 135 days ago

Verified d487007d

make jax as an optional dependency (#9521)

qihqi committed 135 days ago

Verified 83d4253e

Update artifacts_builds.tf for 2.8.0-rc3 wheel (#9527)

bhavya01 committed 136 days ago

Verified c807ebcb

Update artifacts_builds.tf for 2.8.0-rc2 (#9522)

bhavya01 committed 137 days ago

Verified 241cd47f

[XLA] Add placements property to XLAShardedTensor for DTensor compatibility (#9509)

Hoomaaan committed 137 days ago

Verified 95bee8f0

Deduplicate `GetXlaTensors()` function. (#9518)

ysiraichi committed 137 days ago

Verified cb64f4c7

Add w8a8 kernel blocks for Qwen 2.5 7B (#9517)

kyuyeunk committed 138 days ago

Verified 199a9bd5

Dump C++ and Status propagation stacktraces. (#9492)

ysiraichi committed 138 days ago

Verified 7aa466e2

Error Handling: refactor `GetXlaTensor` and related functions to use status types. (#9510)

ysiraichi committed 138 days ago

Verified cd3bd91f

Error Handling: refactor `ExecuteComputation` and `ExecuteReplicated` to propagate status. (#9445)

ysiraichi committed 138 days ago

Verified b0ffc491

Error Handling: propagate status for `ReleaseGilAndTransferData` and `XlaDataToTensors`. (#9431)

ysiraichi committed 138 days ago

Verified 1ed6b46c

Add block sizes for Qwen/Qwen2.5-32B-Instruct (#9516)

vanbasten23 committed 139 days ago

Verified 531c724f

Error Handling: return status value when loading PjRt dynamic plugin. (#9495)

ysiraichi committed 139 days ago

Verified 2820f7c5

Optimize w8a8 kernel vmem limit (#9508)

kyuyeunk committed 142 days ago

Verified 29ae4c76

Move mutable properties of env to thread local, misc changes (#9501)

qihqi committed 143 days ago

Verified 0a1594a8

Clean up quantized matmul condition code (#9506)

kyuyeunk committed 143 days ago

Verified 16b12022

Implement XLAShardedTensor._spec and test (#9488)

aws-cph committed 143 days ago

Verified ca471983

Error Handling: refactor `ComputationClient::TransferFromDevice` to propagate status. (#9429)

ysiraichi committed 143 days ago

Verified 299a16b5

[Bugfix] fix ragged attention kernel auto-tuning table key (#9497)

yaochengji committed 144 days ago

Verified 31c4c2f2

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

vfdev-5 committed 146 days ago

Verified e82631ec

Misc fixes: (#9491)

qihqi committed 147 days ago

Verified 26def0f2

Fix nested stableHLO composite regions (#9385)

Carlomus committed 147 days ago

Verified 55b7d021

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

qihqi committed 149 days ago

Verified 95ba754a

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

qihqi committed 149 days ago

Verified f3c7907f

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

bhavya01 committed 149 days ago

Verified 86a99d75

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

zmelumian972 committed 149 days ago

Verified ca914452

Implement collective gather op (#9435)

bfolie committed 150 days ago

Verified a021bf09

Older