pytorch/xla

Pull Requests Commits

Run torchprime CI only when the pull requests have torchprimeci label

bhavya01 committed 1 year ago

Verified be692190

Error Handling: make `XLATensor::Create()` return status type. (#9544)

ysiraichi committed 1 year ago

Verified 095faec1

Use `TORCH_CHECK()` instead of throwing `std::runtime_error` in `XLA_CHECK*()` macros' implementation. (#9542)

ysiraichi committed 1 year ago

Verified 8c1449f6

Refactor the status error message builder. (#9546)

ysiraichi committed 1 year ago

Verified 57cd41c0

Update artifacts_builds.tf for rc5

bhavya01 committed 1 year ago

Verified 41bfd625

Improve error message of functions related to `GetXlaTensor()`. (#9520)

ysiraichi committed 1 year ago

Verified 6050927c

Update torchbench pin location. (#9543)

ysiraichi committed 1 year ago

Verified 30ad68a2

Add triggers for release 2.8.0 (#9545)

bhavya01 committed 1 year ago

Verified e5e75a84

Do not set `PJRT_DEVICE=CUDA` automatically on import. (#9540)

ysiraichi committed 1 year ago

Verified 15496cd2

Implement XLAShardedTensor.redistribute and test (#9529)

aws-cph committed 1 year ago

Verified 43589c03

Revert 2 accidental commits that I made. (#9536)

qihqi committed 1 year ago

Verified 2889f69e

qihqi committed 1 year ago

b6a5b82b

Add gemini edited docstring

qihqi committed 1 year ago

2ccd5dc9

Optimize KV cache dequantization performance (#9528)

kyuyeunk committed 1 year ago

Verified 9995e971

Remove cuda builds for release wheels (#9533)

bhavya01 committed 1 year ago

Verified d3d91a84

Fix pip install torch_xla[pallas] (#9531)

bhavya01 committed 1 year ago

Verified adf305f3

Update artifacts_builds.tf for 2.8.0-rc4 (#9532)

bhavya01 committed 1 year ago

Verified ebefc8f1

Introduce PlatformVersion bindings (#9513)

rpsilva-aws committed 1 year ago

Verified 0ad39c22

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

vanbasten23 committed 1 year ago

Verified 7a48185d

Reorganize PyTorch/XLA Overview page (#9498)

melissawm committed 1 year ago

Verified d487007d

make jax as an optional dependency (#9521)

qihqi committed 1 year ago

Verified 83d4253e

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

bhavya01 committed 1 year ago

Verified c807ebcb

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

bhavya01 committed 1 year ago

Verified 241cd47f

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

Hoomaaan committed 1 year ago

Verified 95bee8f0

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

ysiraichi committed 1 year ago

Verified cb64f4c7

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

kyuyeunk committed 1 year ago

Verified 199a9bd5

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

ysiraichi committed 1 year ago

Verified 7aa466e2

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

ysiraichi committed 1 year ago

Verified cd3bd91f

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

ysiraichi committed 1 year ago

Verified b0ffc491

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

ysiraichi committed 1 year ago

Verified 1ed6b46c

Older