pytorch/xla

Pull Requests Commits

Improve `torch_xla.compile` documentation (#9194)

sdasgup3 committed 194 days ago

Verified 248a8b33

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

sdasgup3 committed 194 days ago

Verified 513ab7ca

Convert some XLA_CHECKs to fatal errors. (#9263)

zhanyong-wan committed 194 days ago

Verified d43142cb

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

zhanyong-wan committed 194 days ago

Verified 51a3bae2

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

zzzwen committed 194 days ago

Verified 808773a9

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

yaochengji committed 195 days ago

Verified 9871aed5

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

ghpvnist committed 196 days ago

Verified 364bb7eb

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

zhanyong-wan committed 196 days ago

Verified aa44bdea

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

yaoshiang committed 196 days ago

Verified 066e69ed

Fix unspecified behavior on custom calls. (#9247)

ysiraichi committed 196 days ago

Verified f0707217

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

ysiraichi committed 197 days ago

Verified 45bff9b8

Fix docstring indentation error (#9244)

ghpvnist committed 200 days ago

Verified b6d8468c

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

ghpvnist committed 201 days ago

Verified 17401545

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

ghpvnist committed 201 days ago

Verified 0d14c8ae

Configure jobs to use better machine (#9229)

ghpvnist committed 201 days ago

Verified 2cf010bd

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

ysiraichi committed 201 days ago

Verified 6665085b

Add description of prerequisites for running benchmarks (#9230)

haifeng-jin committed 201 days ago

Verified 22a9916d

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

zhanyong-wan committed 202 days ago

Verified 6b00363d

Yho doc cicd attempt 1 (#9233)

yaoshiang committed 202 days ago

Verified 01a235ba

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

tengyifei committed 202 days ago

Verified 855125ee

9082 educate users on mat mul precision (#9103)

yaoshiang committed 202 days ago

Verified 358c8622

Subset of existing, approved PR (#9204)

yaoshiang committed 202 days ago

Verified c7ef4395

Fixed reloading of sharded master weights (#9224)

avizon-aws committed 202 days ago

Verified c0f1e620

Update the OpenXLA pin (#9216)

haifeng-jin committed 202 days ago

Verified ff57e53c

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

bythew3i committed 202 days ago

Verified 82e7ee80

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

zhanyong-wan committed 203 days ago

Verified e28174c9

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

zhanyong-wan committed 203 days ago

Verified eedc8d23

Fix the examples in API_GUIDE (#9213)

zhanyong-wan committed 203 days ago

Verified 7ce8493d

Reflow ci.md to 80 chars (#9151)

tengyifei committed 204 days ago

Verified 334b4d3f

Update ci.md and fix typo (#9150)

tengyifei committed 204 days ago

Verified 0b6349b0

Newer Older