pytorch/xla

Pull Requests Commits

pgmoka committed 154 days ago

49f2e84f

Merge remote-tracking branch 'origin/master' into create-distinction-between-local-and-global-devices

pgmoka committed 155 days ago

d2b1ce11

In pjrt runtime client, raise a Python exception if XLA compilation fails. (#9138)

zhanyong-wan committed 155 days ago

Verified d1ca3c48

[torchax] Added support for bicubic and billinear resampling (#9222)

qihqi committed 155 days ago

Verified 2196d0cf

Run `test_collective_permute.py` on TPU CI. (#9257)

ysiraichi committed 155 days ago

Verified 06c5533d

[Kernel] support kv cache quantization in ragged attention kernel (#9249)

yaochengji committed 155 days ago

Verified 2edcd2e3

Improve `torch_xla.compile` documentation (#9194)

sdasgup3 committed 155 days ago

Verified 248a8b33

Document the difference between tracing time and execution time (#9133)

sdasgup3 committed 155 days ago

Verified 513ab7ca

Convert some XLA_CHECKs to fatal errors. (#9263)

zhanyong-wan committed 155 days ago

Verified d43142cb

Optimize build_developer.sh for the common workflow. (#9262)

zhanyong-wan committed 155 days ago

Verified 51a3bae2

fix method declarations

pgmoka committed 155 days ago

04eddf23

Update README.md with supported Python version (#9239)

zzzwen committed 155 days ago

Verified 808773a9

undo bad rebase

pgmoka committed 155 days ago

994744a7

add new api, and refactor GetNumDevices

pgmoka committed 155 days ago

72b9385a

[Kernel] add group_offset and transpose_rhs support in gmm kernel (#9251)

yaochengji committed 156 days ago

Verified 9871aed5

Add relevant imports to code snippets in amp.md (#9255)

ghpvnist committed 157 days ago

Verified 364bb7eb

Revert "Switch bazel default spawn strategy to `sandboxed`" (#9212)

zhanyong-wan committed 157 days ago

Verified aa44bdea

changed build to mostly match upstream PT, fixed some doc references. (#9248)

yaoshiang committed 157 days ago

Verified 066e69ed

Fix unspecified behavior on custom calls. (#9247)

ysiraichi committed 158 days ago

Verified f0707217

Delete unused tracing functions after codegen migration. (#9240)

ysiraichi committed 159 days ago

Verified 45bff9b8

Fix docstring indentation error (#9244)

ghpvnist committed 162 days ago

Verified b6d8468c

Modify NoneRemover to return modified list/tuple (#9232)

ghpvnist committed 162 days ago

Verified 17401545

Separate pallas test into a separate test shard (#9231)

ghpvnist committed 162 days ago

Verified 0d14c8ae

Configure jobs to use better machine (#9229)

ghpvnist committed 162 days ago

Verified 2cf010bd

Fix + Run `DynamicShapeDetector` tests on CI. (#9075)

ysiraichi committed 162 days ago

Verified 6665085b

Add description of prerequisites for running benchmarks (#9230)

haifeng-jin committed 163 days ago

Verified 22a9916d

Make `run_tests.sh` less spammy. (#9237)

zhanyong-wan committed 163 days ago

Verified 6b00363d

Yho doc cicd attempt 1 (#9233)

yaoshiang committed 163 days ago

Verified 01a235ba

Run all torchax tests with `find` (#9227)

tengyifei committed 163 days ago

Verified 855125ee

9082 educate users on mat mul precision (#9103)

yaoshiang committed 163 days ago

Verified 358c8622

Older