pytorch/xla

Pull Requests Commits

Update README.md

pgmoka committed 180 days ago

Verified 86dc793c

Update README.md

pgmoka committed 181 days ago

Verified e96e39bb

Update README.md

pgmoka committed 183 days ago

Verified 741bad34

Update README.md

pgmoka committed 183 days ago

Verified 61e2d37e

Add lowering for `bitwise_left_shift`. (#8865)

ysiraichi committed 183 days ago

Verified 46a7990f

Update README.md with torch_xla-2.8.0 wheels (#8870)

bhavya01 committed 186 days ago

Verified 8030f059

Support collective matmul optimization in mp (#8855)

yaochengji committed 186 days ago

Verified 2ad5b5d3

fix CPP test build failure in upstream (#8867)

zpcore committed 187 days ago

Verified 760675ad

Avoid destorying the cluster (#8864)

tengyifei committed 188 days ago

Verified c6e55ba5

Fix missing min_tpu_nodes argument (#8863)

tengyifei committed 188 days ago

Verified ece5fe7e

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

tengyifei committed 188 days ago

Verified 6a93dcfb

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

bythew3i committed 189 days ago

Verified 4190fc0e

fix dependency (#8852)

zpcore committed 189 days ago

Verified c7d0b1ef

fix existing bazel link removal (#8848)

zpcore committed 189 days ago

Verified 18cdb17a

Make cxx_abi enable by default (#8844)

zpcore committed 190 days ago

Verified 8f314a5a

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

zpcore committed 190 days ago

Verified 4c5cc420

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

pgmoka committed 190 days ago

Verified 76b0ce50

Add info related to codegen (#8834)

pgmoka committed 190 days ago

Verified 5dd50099

Expose mark_sharding_with_gradients as a public API. (#8826)

iwknow committed 191 days ago

Verified 5caaaee8

Fix for triton pin update. (#8825)

ysiraichi committed 192 days ago

Verified 818c92df

Add build triggers for 2.7-rc1 (#8828)

zpcore committed 192 days ago

Verified fed01200

Revamp SPMD guide (#8807)

tengyifei committed 193 days ago

Verified 5beb6511

Adapt Neuron data type tests (#8835)

rpsilva-aws committed 194 days ago

Verified bd46c0ea

Remove xla_mark_sharding_dynamo_custom_op (#8823)

tengyifei committed 194 days ago

Verified 27d455aa

Support unused arguments in xb.call_jax (#8830)

tengyifei committed 195 days ago

Verified b3f38757

[torchax] Fix functional.max_pool (#8814)

dvhg committed 195 days ago

Verified 0aac10e2

Proper test coverage for Neuron CollectivePermute (#8818)

rpsilva-aws committed 195 days ago

Verified acc4c739

Fix jax_env_context (#8819)

bhavya01 committed 195 days ago

Verified 2e04cfb4

Update MarkShardingFunction to be AOTAutograd traceable (#8821)

tengyifei committed 195 days ago

Verified c969276e

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

lsy323 committed 195 days ago

Verified bdf30f66

Older