pytorch/xla

Pull Requests Commits

yaochengji committed 214 days ago

d2f50c2c

yaochengji committed 214 days ago

9be2310d

yaochengji committed 214 days ago

b0fb6c82

Add block size table for ragged_paged_attention

yaochengji committed 214 days ago

f7dd82ac

Trigger build for 2.6.1 patch (#8938)

zpcore committed 214 days ago

Verified 379ebd5f

Update and rename torch_xla2.yml to torchax.yml (#8936)

tengyifei committed 214 days ago

Verified 9e23ca85

Rewrite scan-based GRU based on nn.GRU (#8914)

iwknow committed 214 days ago

Verified 145c33ae

Fix up pin update (#8935)

tengyifei committed 214 days ago

Verified 980ead54

Include torchax in torch_xla (#8895)

tengyifei committed 215 days ago

Verified bb5c6a0b

pin update (#8908)

lsy323 committed 215 days ago

Verified 6400e169

[ragged-paged-attn] Unify kv strided load to one. (#8929)

bythew3i committed 216 days ago

Verified c044c691

Add heuristic default block sizes for different cases in ragged attention kernel (#8922)

yaochengji committed 216 days ago

Verified 6c3f2313

Clean up deprecated APIs (#8927)

zpcore committed 216 days ago

Verified f1fe8bc0

Use f32 scratch for output so we only need to transfer output with desired dtype back to HBM. (#8924)

vanbasten23 committed 216 days ago

Verified f0881b5a

[torchax] Support linalg.det and linalg.lu_solve (#8872)

dvhg committed 217 days ago

Verified 2d6f57a3

Fix cuda dependency not found (#8903)

zpcore committed 217 days ago

Verified dbb03aa1

Lower `isneginf()`. (#8912)

ysiraichi committed 217 days ago

Verified cba9ff9f

Fix fp8 test (#8909)

lsy323 committed 218 days ago

Verified 985fa188

Add Pallas tutorials to the Pallas guide at torch_xla. (#8904)

vanbasten23 committed 218 days ago

Verified 450b4dce

point build to rc3 (#8907)

zpcore committed 218 days ago

Verified 2999e72b

Use shard_as in scan to ensure that inputs and their gradients have the same sharding (#8879)

tengyifei committed 221 days ago

Verified 6d88c089

[Build] Pin cmake to build PyTorch (#8902)

lsy323 committed 221 days ago

Verified fe3bb7f6

Remove dynamic grid (#8896)

bythew3i committed 222 days ago

Verified b24e6e95

Update cudnn dependency for newer cuda versions (#8894)

zpcore committed 222 days ago

Verified 92873377

8740 add single processing to getting started instructions (#8875)

pgmoka committed 222 days ago

Verified e341ff09

[ragged-paged-attn] Combine k_pages and v_pages on num_kv_head (#8892)

bythew3i committed 223 days ago

Verified 7a3c0518

Encapsulate Mesh invariants (#8882)

rpsilva-aws committed 223 days ago

Verified 96ad8f5f

Avoid unnecessary copy in TensorSource (#8849)

lsy323 committed 223 days ago

Verified 8dc5b496

update Dockerfile config for CI (#8886)

zpcore committed 223 days ago

Verified 3066b034

Increase CPU node count in tpu-ci (#8885)

tengyifei committed 223 days ago

Verified 6a7ce475

Older