pytorch/xla

Pull Requests Commits

manual revert of 9368

bfolie committed 100 days ago

d79f548c

Improve `assume_pure` SPMD functionality (#9360)

tengyifei committed 100 days ago

Verified 0be0fbdf

Add an option to not use dlpack. (#9304)

qihqi committed 101 days ago

Verified 55a75404

docs: fix contribute guide command (#9355)

giacomoni committed 102 days ago

Verified 20899c72

Update ragged paged attention kernel to prevent vmem oom (#9346)

Chenyaaang committed 103 days ago

Verified bea86eec

Split _ops and _decomps. (#9323)

qihqi committed 103 days ago

Verified 3a1ed628

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

zhanyong-wan committed 103 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 106 days ago

Verified f921e5e8

Clean up torchax readme. (#9321)

zhanyong-wan committed 106 days ago

Verified e38d2807

enable_python_dispatcher in some XLA custom passes. (#9312)

laithsakka committed 106 days ago

Verified 9a517a95

Test torchprime from PyTorch/XLA (#9152)

tengyifei committed 106 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 110 days ago

Verified 002f27a8

Sync up jax date with libtpu date (#9296)

tengyifei committed 111 days ago

Verified a687881a

Older