pytorch/xla

Pull Requests Commits

Consolidate if-else

bfolie committed 193 days ago

Verified e8db1163

improved documentation

bfolie committed 199 days ago

448bd65e

remove extra blank line

bfolie committed 201 days ago

25ffd2ff

Merge branch 'master' of https://github.com/pytorch/xla into bfolie/9314-scatter

bfolie committed 201 days ago

21548419

remove scatter from unimplemented list

bfolie committed 201 days ago

04032e83

update l_ref in kernel matrix calculation (#9372)

Chenyaaang committed 201 days ago

Verified 77c40f3f

Avoid recompilation caused by `scan_layers` (#9367)

tengyifei committed 201 days ago

Verified 41b9e452

fix all_gather_into_tensor test and logic (#9332)

bfolie committed 201 days ago

Verified 9b0b02f0

generalize implementation to work on longer lists

bfolie committed 201 days ago

fcf2803f

Improve `assume_pure` SPMD functionality (#9360)

tengyifei committed 201 days ago

Verified 0be0fbdf

format, fix small issues

bfolie committed 201 days ago

4940e19d

write first draft of scatter implementation and test

bfolie committed 201 days ago

af92456c

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

qihqi committed 202 days ago

Verified 55a75404

docs: fix contribute guide command (#9355)

giacomoni committed 203 days ago

Verified 20899c72

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

Chenyaaang committed 204 days ago

Verified bea86eec

Split _ops and _decomps. (#9323)

qihqi committed 205 days ago

Verified 3a1ed628

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

zhanyong-wan committed 205 days ago

Verified 15366e94

Switch XLA_CHECKs to ABSL_CHECKs in lowering_context. (#9338)

zhanyong-wan committed 205 days ago

Verified c2aa21ee

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

rickeylev committed 205 days ago

Verified 1d409399

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

bfolie committed 205 days ago

Verified 49b0b3b3

Surface CUDA deprecation warning. (#9333)

ysiraichi committed 205 days ago

Verified edd38caa

Update pins (#9330)

pgmoka committed 206 days ago

Verified 503f8280

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

aws-yyjau committed 206 days ago

Verified 3bc9ece7

Update submodules whenever we switch branches. (#9320)

zhanyong-wan committed 207 days ago

Verified 1bd3042f

Remove nightly CUDA builds (#9329)

tengyifei committed 207 days ago

Verified ed14590a

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

ghpvnist committed 207 days ago

Verified 78cff036

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

ghpvnist committed 207 days ago

Verified ae4f06b3

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

bfolie committed 208 days ago

Verified f921e5e8

Clean up torchax readme. (#9321)

zhanyong-wan committed 208 days ago

Verified e38d2807

enable_python_dispatcher in some XLA custom passes. (#9312)

laithsakka committed 208 days ago

Verified 9a517a95

Older