pytorch/xla

Pull Requests Commits

yaochengji committed 1 year ago

80412afe

yaochengji committed 1 year ago

070138a4

yaochengji committed 1 year ago

5749e4c6

yaochengji committed 1 year ago

52a00099

modify python op

yaochengji committed 1 year ago

d3655f89

[Kernel] support kv cache quantization in ragged attention kernel

yaochengji committed 1 year ago

b57e4578

Fix unspecified behavior on custom calls. (#9247)

ysiraichi committed 1 year ago

Verified f0707217

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

ysiraichi committed 1 year ago

Verified 45bff9b8

Fix docstring indentation error (#9244)

ghpvnist committed 1 year ago

Verified b6d8468c

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

ghpvnist committed 1 year ago

Verified 17401545

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

ghpvnist committed 1 year ago

Verified 0d14c8ae

Configure jobs to use better machine (#9229)

ghpvnist committed 1 year ago

Verified 2cf010bd

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

ysiraichi committed 1 year ago

Verified 6665085b

Add description of prerequisites for running benchmarks (#9230)

haifeng-jin committed 1 year ago

Verified 22a9916d

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

zhanyong-wan committed 1 year ago

Verified 6b00363d

Yho doc cicd attempt 1 (#9233)

yaoshiang committed 1 year ago

Verified 01a235ba

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

tengyifei committed 1 year ago

Verified 855125ee

9082 educate users on mat mul precision (#9103)

yaoshiang committed 1 year ago

Verified 358c8622

Subset of existing, approved PR (#9204)

yaoshiang committed 1 year ago

Verified c7ef4395

Fixed reloading of sharded master weights (#9224)

avizon-aws committed 1 year ago

Verified c0f1e620

Update the OpenXLA pin (#9216)

haifeng-jin committed 1 year ago

Verified ff57e53c

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

bythew3i committed 1 year ago

Verified 82e7ee80

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

zhanyong-wan committed 1 year ago

Verified e28174c9

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

zhanyong-wan committed 1 year ago

Verified eedc8d23

Fix the examples in API_GUIDE (#9213)

zhanyong-wan committed 1 year ago

Verified 7ce8493d

Reflow ci.md to 80 chars (#9151)

tengyifei committed 1 year ago

Verified 334b4d3f

Update ci.md and fix typo (#9150)

tengyifei committed 1 year ago

Verified 0b6349b0

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

zhanyong-wan committed 1 year ago

Verified 6e3f4076

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

ghpvnist committed 1 year ago

Verified b5937198

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

zhanyong-wan committed 1 year ago

Verified 01b5408d

Older