pytorch/xla

Pull Requests Commits

delete setup.py

ManfeiBai committed 211 days ago

c1f537f5

Update setup.py

ManfeiBai committed 211 days ago

Verified d52d10cd

Create pyproject.toml

ManfeiBai committed 211 days ago

Verified 053cfbd6

Create fix_pyproject.py

ManfeiBai committed 211 days ago

Verified ffe81f62

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

tengyifei committed 212 days ago

Verified 6f020aad

Add padding ragged paged attention test (#8741)

vanbasten23 committed 212 days ago

Verified 5e1478d6

Improve the ragged kernel benchmarking script. (#8733)

vanbasten23 committed 212 days ago

Verified 79d1e86b

Add autograd function for mark_sharding (#8723)

bhavya01 committed 217 days ago

Verified 1acc987a

Add information to xla.launch (#8724)

pgmoka committed 217 days ago

Verified a08b7f86

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

vanbasten23 committed 217 days ago

Verified 0c4e82d6

Bump CUDNN version in docker image. (#8718)

ysiraichi committed 218 days ago

Verified 60160233

Bump XLA commit pin. (#8714)

ysiraichi committed 219 days ago

Verified ca248c7e

Support splitting physical axis in SPMD mesh (#8698)

tengyifei committed 222 days ago

Verified 8a712495

update pin (#8677)

lsy323 committed 224 days ago

Verified 6f423d0b

Fix CUDA plugin CI. (#8593)

ysiraichi committed 224 days ago

Verified 42edbe18

Include the alias ID in GetXLATensorDebugInfo (#8701)

rpsilva-aws committed 225 days ago

Verified 3a4ad6f3

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

bythew3i committed 225 days ago

Verified 06e1b591

Mention remote build in CONTRIBUTING.md (#8699)

tengyifei committed 225 days ago

Verified 45af5193

Add missing NeuronPlugin configure_single_process function (#8694)

jeffhataws committed 226 days ago

Verified 065cb5b1

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

zhangp365 committed 226 days ago

Verified cff9f4e0

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

tengyifei committed 229 days ago

Verified c0afda3a

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

lsy323 committed 229 days ago

Verified ace5e24b

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

zmelumian972 committed 229 days ago

Verified 0cd1fc2a

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

avizon-aws committed 230 days ago

Verified afaf0d05

Misc fixes (#8679)

qihqi committed 231 days ago

Verified 8b45e599

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

yaochengji committed 231 days ago

Verified 45109d1e

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

hosseinsarshar committed 231 days ago

Verified b9bef193

fix a bug in tensor.flatten (#8680)

yaochengji committed 232 days ago

Verified 49cd3105

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

ysiraichi committed 232 days ago

Verified ec10537c

[CI] Fix torchbench setup failure (#8670)

lsy323 committed 233 days ago

Verified a4356250

Older