pytorch/xla

Pull Requests Commits

Add start_trace and stop_trace API in profiler (#8743)

lsy323 committed 251 days ago

Verified b4ba17b5

Add --network=host to TPU docker build command (#8735)

bhavya01 committed 251 days ago

Verified 71d0ce65

Increase tolerance in test_scan_xla_patched_linear (#8749)

tengyifei committed 251 days ago

Verified ac430abf

Extend buffer donation aliasing APIs (#8721)

rpsilva-aws committed 251 days ago

Verified 07976424

Introduce apply_xla_patch_to_nn_linear and test that in a scan (#8739)

tengyifei committed 252 days ago

Verified 6f020aad

Add padding ragged paged attention test (#8741)

vanbasten23 committed 252 days ago

Verified 5e1478d6

Improve the ragged kernel benchmarking script. (#8733)

vanbasten23 committed 253 days ago

Verified 79d1e86b

Add autograd function for mark_sharding (#8723)

bhavya01 committed 257 days ago

Verified 1acc987a

Add information to xla.launch (#8724)

pgmoka committed 257 days ago

Verified a08b7f86

Split page indices in the ragged paged attention. (#8688)

vanbasten23 committed 258 days ago

Verified 0c4e82d6

Bump CUDNN version in docker image. (#8718)

ysiraichi committed 259 days ago

Verified 60160233

Bump XLA commit pin. (#8714)

ysiraichi committed 260 days ago

Verified ca248c7e

Support splitting physical axis in SPMD mesh (#8698)

tengyifei committed 263 days ago

Verified 8a712495

update pin (#8677)

lsy323 committed 264 days ago

Verified 6f423d0b

Fix CUDA plugin CI. (#8593)

ysiraichi committed 265 days ago

Verified 42edbe18

Include the alias ID in GetXLATensorDebugInfo (#8701)

rpsilva-aws committed 265 days ago

Verified 3a4ad6f3

Integrate ragged paged attn to pytorch/xla (#8692)

bythew3i committed 266 days ago

Verified 06e1b591

Mention remote build in CONTRIBUTING.md (#8699)

tengyifei committed 266 days ago

Verified 45af5193

Add missing NeuronPlugin configure_single_process function (#8694)

jeffhataws committed 267 days ago

Verified 065cb5b1

Fix a bug in flash attention where kv_seq_len should divide block_k_major. (#8671)

zhangp365 committed 267 days ago

Verified cff9f4e0

Add another C++11 ABI docker reference (#8649)

tengyifei committed 270 days ago

Verified c0afda3a

[torchax] Update installation guide in readme (#8664)

lsy323 committed 270 days ago

Verified ace5e24b

Fix dk/dv autograd error on TPU flash attention (#8685)

zmelumian972 committed 270 days ago

Verified 0cd1fc2a

Added pow to autocast policy and unit test (#8684)

avizon-aws committed 270 days ago

Verified afaf0d05

Misc fixes (#8679)

qihqi committed 271 days ago

Verified 8b45e599

Fix aten.linear and add aten.einsum (#8666)

yaochengji committed 272 days ago

Verified 45109d1e

fixed the type miss match issue with symint in SPMD regime and dynamo (#8681)

hosseinsarshar committed 272 days ago

Verified b9bef193

fix a bug in tensor.flatten (#8680)

yaochengji committed 273 days ago

Verified 49cd3105

Install `clang-17` on development Dockerfile. (#8673)

ysiraichi committed 273 days ago

Verified ec10537c

[CI] Fix torchbench setup failure (#8670)

lsy323 committed 274 days ago

Verified a4356250

Newer Older