pytorch/xla

Pull Requests Commits

yaochengji committed 1 year ago

d2f50c2c

yaochengji committed 1 year ago

9be2310d

yaochengji committed 1 year ago

b0fb6c82

Add block size table for ragged_paged_attention

yaochengji committed 1 year ago

f7dd82ac

Trigger build for 2.6.1 patch (#8938)

zpcore committed 1 year ago

Verified 379ebd5f

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

tengyifei committed 1 year ago

Verified 9e23ca85

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

iwknow committed 1 year ago

Verified 145c33ae

Fix up pin update (#8935)

tengyifei committed 1 year ago

Verified 980ead54

Include torchax in torch_xla (#8895)

tengyifei committed 1 year ago

Verified bb5c6a0b

pin update (#8908)

lsy323 committed 1 year ago

Verified 6400e169

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

bythew3i committed 1 year ago

Verified c044c691

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

yaochengji committed 1 year ago

Verified 6c3f2313

Clean up deprecated APIs (#8927)

zpcore committed 1 year ago

Verified f1fe8bc0

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

vanbasten23 committed 1 year ago

Verified f0881b5a

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

dvhg committed 1 year ago

Verified 2d6f57a3

Fix cuda dependency not found (#8903)

zpcore committed 1 year ago

Verified dbb03aa1

Lower `isneginf()`. (#8912)

ysiraichi committed 1 year ago

Verified cba9ff9f

Fix fp8 test (#8909)

lsy323 committed 1 year ago

Verified 985fa188

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

vanbasten23 committed 1 year ago

Verified 450b4dce

point build to rc3 (#8907)

zpcore committed 1 year ago

Verified 2999e72b

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

tengyifei committed 1 year ago

Verified 6d88c089

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

lsy323 committed 1 year ago

Verified fe3bb7f6

Remove dynamic grid (#8896)

bythew3i committed 1 year ago

Verified b24e6e95

Update cudnn dependency for newer cuda versions (#8894)

zpcore committed 1 year ago

Verified 92873377

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

pgmoka committed 1 year ago

Verified e341ff09

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

bythew3i committed 1 year ago

Verified 7a3c0518

Encapsulate Mesh invariants (#8882)

rpsilva-aws committed 1 year ago

Verified 96ad8f5f

Avoid unnecessary copy in TensorSource (#8849)

lsy323 committed 1 year ago

Verified 8dc5b496

update Dockerfile config for CI (#8886)

zpcore committed 1 year ago

Verified 3066b034

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

tengyifei committed 1 year ago

Verified 6a7ce475

Older