pytorch/xla

Pull Requests Commits

Ignore the MODULE.bazel* files generated by bazel. (#9352)

zhanyong-wan committed 104 days ago

Verified 15366e94

Switch XLA_CHECKs to ABSL_CHECKs in lowering_context. (#9338)

zhanyong-wan committed 104 days ago

Verified c2aa21ee

cleanup: remove defunct --distinct_host_configuration flag (#9300)

rickeylev committed 104 days ago

Verified 1d409399

Fix usage of dimensions_size() to check for tuple (#9347)

bfolie committed 104 days ago

Verified 49b0b3b3

Surface CUDA deprecation warning. (#9333)

ysiraichi committed 104 days ago

Verified edd38caa

Update pins (#9330)

pgmoka committed 105 days ago

Verified 503f8280

doc: update pytorch-on-xla-devices and troubleshoot doc for tensor synchronization issue (#9258)

aws-yyjau committed 105 days ago

Verified 3bc9ece7

Update submodules whenever we switch branches. (#9320)

zhanyong-wan committed 106 days ago

Verified 1bd3042f

Remove nightly CUDA builds (#9329)

tengyifei committed 106 days ago

Verified ed14590a

Migrate `.to(torch_xla.device())` to `.to('xla')` (#9324)

ghpvnist committed 106 days ago

Verified 78cff036

Migrate uses of `import torch_xla as xla` to `import torch_xla` (#9325)

ghpvnist committed 106 days ago

Verified ae4f06b3

Revert Create copy of nightly wheel without version number (#9318)

bfolie committed 107 days ago

Verified f921e5e8

Clean up torchax readme. (#9321)

zhanyong-wan committed 107 days ago

Verified e38d2807

enable_python_dispatcher in some XLA custom passes. (#9312)

laithsakka committed 107 days ago

Verified 9a517a95

Test torchprime from PyTorch/XLA (#9152)

tengyifei committed 107 days ago

Verified 728043e6

Extend build_developer.sh and git_sync_main.py with a -a flag. (#9319)

zhanyong-wan committed 107 days ago

Verified 37ed591d

Fix torchgen imports in codegen (#9310)

XuehaiPan committed 107 days ago

Verified d82e15c4

Stop tracking data and log files (#9308)

zzzwen committed 107 days ago

Verified 7875be0f

Support trace_me and xp.Trace in assume_pure (#9311)

tengyifei committed 107 days ago

Verified d663c7ed

Update dependency Updater (#9302)

tengyifei committed 110 days ago

Verified 6e8e7dbe

Merge cpp_tests1 and cpp_tests2 into single shard (#9298)

ghpvnist committed 110 days ago

Verified 6ed3ca15

Add a script to automatically format C++/Python files upon `git push` (#9293)

zhanyong-wan committed 110 days ago

Verified 9166877e

[torchax] Fixes test for kthvalue pytorch/xla#7458 (#9223)

qianminj123 committed 111 days ago

Verified 002f27a8

Sync up jax date with libtpu date (#9296)

tengyifei committed 111 days ago

Verified a687881a

[dynamic shapes] support sym_min, sym_max for XLA SymNode (#9291)

pianpwk committed 111 days ago

Verified e408a33c

Misc torchax improvements (#9294)

qihqi committed 111 days ago

Verified 6bab9e2c

Do not clear UncachedCompile between graph executions (#9282)

haifeng-jin committed 111 days ago

Verified c7a96713

Pin update 06052025 (#9299)

bhavya01 committed 111 days ago

Verified 01072eb2

Deprecate XLA:CUDA: add warning on initialization. (#9295)

ysiraichi committed 111 days ago

Verified 70de5681

Split CPU tests so the shards have a more even test time (#9292)

ghpvnist committed 111 days ago

Verified 01decbbd

Newer Older