pytorch/xla

Pull Requests Commits

remove unnessary astype

yaochengji committed 357 days ago

e7bd5f2d

yaochengji committed 357 days ago

04ceed5a

yaochengji committed 358 days ago

1120fcde

yaochengji committed 358 days ago

5c690075

Revert "Change num_seqs type from int to torch.Tensor" (#8767)

yaochengji committed 358 days ago

Verified 2e4f073a

Enable default buffer donation for gradient accumulation (#8758)

rpsilva-aws committed 358 days ago

Verified 6712eb91

Change num_seqs type from int to torch.Tensor (#8736)

vanbasten23 committed 358 days ago

Verified b18a65fa

Lower `as_strided_copy` use fast path with `slice` (#8734)

zpcore committed 359 days ago

Verified 1ab8216a

Add start_trace and stop_trace API in profiler (#8743)

lsy323 committed 359 days ago

Verified b4ba17b5

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

bhavya01 committed 359 days ago

Verified 71d0ce65

Increase tolerance in test_scan_xla_patched_linear (#8749)

tengyifei committed 359 days ago

Verified ac430abf

Extend buffer donation aliasing APIs (#8721)

rpsilva-aws committed 359 days ago

Verified 07976424

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

tengyifei committed 360 days ago

Verified 6f020aad

Add padding ragged paged attention test (#8741)

vanbasten23 committed 360 days ago

Verified 5e1478d6

Improve the ragged kernel benchmarking script. (#8733)

vanbasten23 committed 361 days ago

Verified 79d1e86b

Add autograd function for mark_sharding (#8723)

bhavya01 committed 1 year ago

Verified 1acc987a

Add information to xla.launch (#8724)

pgmoka committed 1 year ago

Verified a08b7f86

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

vanbasten23 committed 1 year ago

Verified 0c4e82d6

Bump CUDNN version in docker image. (#8718)

ysiraichi committed 1 year ago

Verified 60160233

Bump XLA commit pin. (#8714)

ysiraichi committed 1 year ago

Verified ca248c7e

Support splitting physical axis in SPMD mesh (#8698)

tengyifei committed 1 year ago

Verified 8a712495

update pin (#8677)

lsy323 committed 1 year ago

Verified 6f423d0b

Fix CUDA plugin CI. (#8593)

ysiraichi committed 1 year ago

Verified 42edbe18

Include the alias ID in GetXLATensorDebugInfo (#8701)

rpsilva-aws committed 1 year ago

Verified 3a4ad6f3

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

bythew3i committed 1 year ago

Verified 06e1b591

Mention remote build in CONTRIBUTING.md (#8699)

tengyifei committed 1 year ago

Verified 45af5193

Add missing NeuronPlugin configure_single_process function (#8694)

jeffhataws committed 1 year ago

Verified 065cb5b1

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

zhangp365 committed 1 year ago

Verified cff9f4e0

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

tengyifei committed 1 year ago

Verified c0afda3a

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

lsy323 committed 1 year ago

Verified ace5e24b

Older