pytorch/xla

Pull Requests Commits

yaochengji committed 230 days ago

80412afe

yaochengji committed 231 days ago

070138a4

yaochengji committed 231 days ago

5749e4c6

yaochengji committed 231 days ago

52a00099

modify python op

yaochengji committed 231 days ago

d3655f89

[Kernel] support kv cache quantization in ragged attention kernel

yaochengji committed 231 days ago

b57e4578

Fix unspecified behavior on custom calls. (#9247)

ysiraichi committed 231 days ago

Verified f0707217

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

ysiraichi committed 232 days ago

Verified 45bff9b8

Fix docstring indentation error (#9244)

ghpvnist committed 235 days ago

Verified b6d8468c

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

ghpvnist committed 236 days ago

Verified 17401545

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

ghpvnist committed 236 days ago

Verified 0d14c8ae

Configure jobs to use better machine (#9229)

ghpvnist committed 236 days ago

Verified 2cf010bd

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

ysiraichi committed 236 days ago

Verified 6665085b

Add description of prerequisites for running benchmarks (#9230)

haifeng-jin committed 236 days ago

Verified 22a9916d

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

zhanyong-wan committed 237 days ago

Verified 6b00363d

Yho doc cicd attempt 1 (#9233)

yaoshiang committed 237 days ago

Verified 01a235ba

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

tengyifei committed 237 days ago

Verified 855125ee

9082 educate users on mat mul precision (#9103)

yaoshiang committed 237 days ago

Verified 358c8622

Subset of existing, approved PR (#9204)

yaoshiang committed 237 days ago

Verified c7ef4395

Fixed reloading of sharded master weights (#9224)

avizon-aws committed 237 days ago

Verified c0f1e620

Update the OpenXLA pin (#9216)

haifeng-jin committed 237 days ago

Verified ff57e53c

[ragged-paged-attn] Apply kv mask to filter out NaNs (#9219)

bythew3i committed 237 days ago

Verified 82e7ee80

Document test/neuron/run_tests.sh. (#9205)

zhanyong-wan committed 238 days ago

Verified e28174c9

Improve CI build speed by letting bazel decide how many worker threads to have (#9209)

zhanyong-wan committed 238 days ago

Verified eedc8d23

Fix the examples in API_GUIDE (#9213)

zhanyong-wan committed 238 days ago

Verified 7ce8493d

Reflow ci.md to 80 chars (#9151)

tengyifei committed 239 days ago

Verified 334b4d3f

Update ci.md and fix typo (#9150)

tengyifei committed 239 days ago

Verified 0b6349b0

Clean up env var usages in run_tests.sh scripts. (#9207)

zhanyong-wan committed 239 days ago

Verified 6e3f4076

Split expensive TPU tests to run in parallel (#9198)

ghpvnist committed 239 days ago

Verified b5937198

Make 2 more run_tests.sh support test selection (#9192)

zhanyong-wan committed 239 days ago

Verified 01b5408d

Older