pytorch/xla

Pull Requests Commits

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

tengyifei committed 220 days ago

Verified 6a7ce475

Cache HLO in xb.call_jax and support non-tensor args (#8878)

tengyifei committed 222 days ago

Verified a3ef52e5

Add lowering for `bitwise_right_shift`. (#8866)

ysiraichi committed 222 days ago

Verified 2aa11ccc

Correct type conversions in OpBuilder and add corresponding test. (#8873)

iwknow committed 222 days ago

Verified f2bdecfa

8727 create a site map or centralize links in readme (#8843)

pgmoka committed 222 days ago

Verified 3808df01

Add lowering for `bitwise_left_shift`. (#8865)

ysiraichi committed 222 days ago

Verified 46a7990f

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

bhavya01 committed 225 days ago

Verified 8030f059

Support collective matmul optimization in mp (#8855)

yaochengji committed 225 days ago

Verified 2ad5b5d3

fix CPP test build failure in upstream (#8867)

zpcore committed 226 days ago

Verified 760675ad

Avoid destorying the cluster (#8864)

tengyifei committed 227 days ago

Verified c6e55ba5

Fix missing min_tpu_nodes argument (#8863)

tengyifei committed 227 days ago

Verified ece5fe7e

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

tengyifei committed 227 days ago

Verified 6a93dcfb

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

bythew3i committed 228 days ago

Verified 4190fc0e

fix dependency (#8852)

zpcore committed 228 days ago

Verified c7d0b1ef

fix existing bazel link removal (#8848)

zpcore committed 228 days ago

Verified 18cdb17a

Make cxx_abi enable by default (#8844)

zpcore committed 229 days ago

Verified 8f314a5a

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

zpcore committed 229 days ago

Verified 4c5cc420

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

pgmoka committed 229 days ago

Verified 76b0ce50

Add info related to codegen (#8834)

pgmoka committed 229 days ago

Verified 5dd50099

Expose mark_sharding_with_gradients as a public API. (#8826)

iwknow committed 230 days ago

Verified 5caaaee8

Fix for triton pin update. (#8825)

ysiraichi committed 231 days ago

Verified 818c92df

Add build triggers for 2.7-rc1 (#8828)

zpcore committed 231 days ago

Verified fed01200

Revamp SPMD guide (#8807)

tengyifei committed 232 days ago

Verified 5beb6511

Adapt Neuron data type tests (#8835)

rpsilva-aws committed 233 days ago

Verified bd46c0ea

Remove xla_mark_sharding_dynamo_custom_op (#8823)

tengyifei committed 233 days ago

Verified 27d455aa

Support unused arguments in xb.call_jax (#8830)

tengyifei committed 234 days ago

Verified b3f38757

[torchax] Fix functional.max_pool (#8814)

dvhg committed 234 days ago

Verified 0aac10e2

Proper test coverage for Neuron CollectivePermute (#8818)

rpsilva-aws committed 234 days ago

Verified acc4c739

Fix jax_env_context (#8819)

bhavya01 committed 234 days ago

Verified 2e04cfb4

Update MarkShardingFunction to be AOTAutograd traceable (#8821)

tengyifei committed 234 days ago

Verified c969276e

Newer Older