pytorch/xla

Pull Requests Commits

`clamp`: improve error handling and error messages. (#9642)

ysiraichi committed 140 days ago

Verified 87e631a6

Add Shardy support for Torch-XLA (#9541)

hshahTT committed 144 days ago

Verified 0fa6e312

feat: add support for custom compile options in torch_xla.compile and… (#9575)

sshonTT committed 146 days ago

Verified b1131d1d

Remove python 3.9 wheels and update terraform PyTorch/XLA release version to 2.9 (#9673)

bhavya01 committed 147 days ago

Verified 2619a3f6

Remove torchax from torchxla repo. (#9672)

qihqi committed 148 days ago

Verified eb37dfdf

Update PyTorch and XLA pin. (#9668)

ysiraichi committed 148 days ago

Verified 11590c1c

mul: remove opmath cast sequence (#9663)

sshonTT committed 152 days ago

Verified 2a9138a2

Accept conda channels' ToS with environment variable. (#9666)

ysiraichi committed 153 days ago

Verified 1ab67879

fix(xla): convert group-local to global ranks in broadcast (#9657)

Hoomaaan committed 153 days ago

Verified 13485454

Bump Python version of `ci-tpu-test-trigger` to 3.12. (#9665)

ysiraichi committed 154 days ago

Verified 420adaa4

Revert "Fix Terraform usage of `cuda_version`. (#9655)" (#9664)

ysiraichi committed 154 days ago

Verified cc300f79

Accept conda channels' ToS when building the upstream docker image. (#9661)

ysiraichi committed 154 days ago

Verified 6ac4a7cd

Create PyTorch commit pin. (#9654)

ysiraichi committed 154 days ago

Verified 3862b878

Fix Terraform usage of `cuda_version`. (#9655)

ysiraichi committed 154 days ago

Verified 32401663

`trace`: improve error handling and error messages. (#9630)

ysiraichi committed 154 days ago

Verified a5116917

Pin `flax` and skip C++ test `SiLUBackward`. (#9660)

ysiraichi committed 154 days ago

Verified 302c3f1f

Add default args for _aten_conv2d (#9623)

hsjts0u committed 155 days ago

Verified 03d4dc06

update gcc (#9650)

qihqi committed 160 days ago

Verified 0fc62aa2

`expand`: improve error handling and error messages. (#9645)

ysiraichi committed 168 days ago

Verified 0c0ae2d2

`stack`: improve error handling and error messages. (#9629)

ysiraichi committed 169 days ago

Verified 6d755eec

`roll`: improve error handling and error messages. (#9628)

ysiraichi committed 169 days ago

Verified a66cfc38

Do not warn on jax usage when workarounds are available (#9624)

bhavya01 committed 170 days ago

Verified efe20ab8

Fix `test_ops_error_message.py` and run it on CI. (#9640)

ysiraichi committed 170 days ago

Verified 23297469

Move torch ops error message tests into a new file. (#9622)

ysiraichi committed 174 days ago

Verified c77852e1

Set `allow_broken_conditionals` configuration variable at `ansible.cfg`. (#9634)

ysiraichi committed 174 days ago

Verified 8efa5682

support load and save checkpoint in torchax (#9616)

junjieqian committed 175 days ago

Verified 7aba922c

Remove libopenblas-dev from ansible dependencies (#9632)

bhavya01 committed 175 days ago

Verified caa809f7

Upgrade build infra to use debian-12 and gcc-11 (#9631)

bhavya01 committed 175 days ago

Verified 49dec2ee

Replace `GetComputationClientOrDie()` with `GetComputationClient()` (part 2). (#9620)

ysiraichi committed 176 days ago

Verified aba96d81

Add triggers for v2.8.1 version

bhavya01 committed 179 days ago

Verified 6c5478ff

Newer Older