pytorch/xla

Pull Requests Commits

Support collective matmul optimization in mp (#8855)

yaochengji committed 226 days ago

Verified 2ad5b5d3

fix CPP test build failure in upstream (#8867)

zpcore committed 227 days ago

Verified 760675ad

Avoid destorying the cluster (#8864)

tengyifei committed 228 days ago

Verified c6e55ba5

Fix missing min_tpu_nodes argument (#8863)

tengyifei committed 228 days ago

Verified ece5fe7e

Increase TPU CI node count from 2 to 32 (#8857)

tengyifei committed 228 days ago

Verified 6a93dcfb

[ragged-paged-attn] Use hidden states in kv cache and support any num_kv_head (#8851)

bythew3i committed 229 days ago

Verified 4190fc0e

fix dependency (#8852)

zpcore committed 229 days ago

Verified c7d0b1ef

fix existing bazel link removal (#8848)

zpcore committed 229 days ago

Verified 18cdb17a

Make cxx_abi enable by default (#8844)

zpcore committed 230 days ago

Verified 8f314a5a

install jax dependency by default to unblock pytorch ci (#8845)

zpcore committed 230 days ago

Verified 4c5cc420

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

pgmoka committed 230 days ago

Verified 76b0ce50

Add info related to codegen (#8834)

pgmoka committed 230 days ago

Verified 5dd50099

Expose mark_sharding_with_gradients as a public API. (#8826)

iwknow committed 231 days ago

Verified 5caaaee8

Fix for triton pin update. (#8825)

ysiraichi committed 232 days ago

Verified 818c92df

Add build triggers for 2.7-rc1 (#8828)

zpcore committed 232 days ago

Verified fed01200

Revamp SPMD guide (#8807)

tengyifei committed 233 days ago

Verified 5beb6511

Adapt Neuron data type tests (#8835)

rpsilva-aws committed 234 days ago

Verified bd46c0ea

Remove xla_mark_sharding_dynamo_custom_op (#8823)

tengyifei committed 234 days ago

Verified 27d455aa

Support unused arguments in xb.call_jax (#8830)

tengyifei committed 235 days ago

Verified b3f38757

[torchax] Fix functional.max_pool (#8814)

dvhg committed 235 days ago

Verified 0aac10e2

Proper test coverage for Neuron CollectivePermute (#8818)

rpsilva-aws committed 235 days ago

Verified acc4c739

Fix jax_env_context (#8819)

bhavya01 committed 235 days ago

Verified 2e04cfb4

Update MarkShardingFunction to be AOTAutograd traceable (#8821)

tengyifei committed 235 days ago

Verified c969276e

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

lsy323 committed 235 days ago

Verified bdf30f66

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

qihqi committed 236 days ago

Verified 53b3ab81

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

bythew3i committed 236 days ago

Verified af16c210

Fix the ragged kernel bug (#8806)

vanbasten23 committed 237 days ago

Verified 6804c65f

Document xla override (#8802)

pgmoka committed 238 days ago

Verified 0407d6d1

Leverage placeholder tensors for GA API (#8808)

rpsilva-aws committed 238 days ago

Verified 5b7462c6

Manually register einsum on xla (#8801)

pgmoka committed 241 days ago

Verified 1db6c861

Newer Older