pytorch/xla

Pull Requests Commits

Update README.md

pgmoka committed 1 year ago

Verified 9baaa543

Update documentation.md

pgmoka committed 1 year ago

Verified 307eb1ff

Update README.md

pgmoka committed 1 year ago

Verified 9bd75685

Add information about codegen in codegen/xla_native_functions.yaml (#8817)

pgmoka committed 1 year ago

Verified 76b0ce50

Add info related to codegen (#8834)

pgmoka committed 1 year ago

Verified 5dd50099

Expose mark_sharding_with_gradients as a public API. (#8826)

iwknow committed 1 year ago

Verified 5caaaee8

Fix for triton pin update. (#8825)

ysiraichi committed 1 year ago

Verified 818c92df

Add build triggers for 2.7-rc1 (#8828)

zpcore committed 1 year ago

Verified fed01200

Revamp SPMD guide (#8807)

tengyifei committed 1 year ago

Verified 5beb6511

Adapt Neuron data type tests (#8835)

rpsilva-aws committed 1 year ago

Verified bd46c0ea

Remove xla_mark_sharding_dynamo_custom_op (#8823)

tengyifei committed 1 year ago

Verified 27d455aa

Support unused arguments in xb.call_jax (#8830)

tengyifei committed 1 year ago

Verified b3f38757

[torchax] Fix functional.max_pool (#8814)

dvhg committed 1 year ago

Verified 0aac10e2

Proper test coverage for Neuron CollectivePermute (#8818)

rpsilva-aws committed 1 year ago

Verified acc4c739

Fix jax_env_context (#8819)

bhavya01 committed 1 year ago

Verified 2e04cfb4

Update MarkShardingFunction to be AOTAutograd traceable (#8821)

tengyifei committed 1 year ago

Verified c969276e

[Compilation Cache] Add API to query number of cached computation graphs (#8822)

lsy323 committed 1 year ago

Verified bdf30f66

Refactor Computation of jax into creation and call (#8789)

qihqi committed 1 year ago

Verified 53b3ab81

Remove pages_per_seq padding constrains and add MHA support (#8816)

bythew3i committed 1 year ago

Verified af16c210

Fix the ragged kernel bug (#8806)

vanbasten23 committed 1 year ago

Verified 6804c65f

Document xla override (#8802)

pgmoka committed 1 year ago

Verified 0407d6d1

Leverage placeholder tensors for GA API (#8808)

rpsilva-aws committed 1 year ago

Verified 5b7462c6

Manually register einsum on xla (#8801)

pgmoka committed 1 year ago

Verified 1db6c861

Misc fixes (#8804)

qihqi committed 1 year ago

Verified 7aea0585

Re-enable triton tests. (#8803)

ysiraichi committed 1 year ago

Verified 2c70a1c2

Lower `_conj_copy` and `alias` operation. (#8686)

ysiraichi committed 1 year ago

Verified 6c53a1ef

Fix ragged paged attention v2 to resolve the recompilation issue (#8797)

yaochengji committed 1 year ago

Verified d06a9c97

Pin update to 20250303 (#8788)

tengyifei committed 1 year ago

Verified 9b61c1ad

Use placeholder tensor in scan (#8785)

tengyifei committed 1 year ago

Verified a9279287

Use .backward() with in-place grad mutations for the GA API (#8768)

rpsilva-aws committed 1 year ago

Verified 4540d817

Older