pytorch/xla

Pull Requests Commits

Fix fp8 test (#8909)

lsy323 committed 327 days ago

Verified 985fa188

Add Pallas tutorials to the Pallas guide at torch_xla. (#8904)

vanbasten23 committed 327 days ago

Verified 450b4dce

point build to rc3 (#8907)

zpcore committed 328 days ago

Verified 2999e72b

Use shard_as in scan to ensure that inputs and their gradients have the same sharding (#8879)

tengyifei committed 330 days ago

Verified 6d88c089

[Build] Pin cmake to build PyTorch (#8902)

lsy323 committed 331 days ago

Verified fe3bb7f6

Remove dynamic grid (#8896)

bythew3i committed 331 days ago

Verified b24e6e95

Update cudnn dependency for newer cuda versions (#8894)

zpcore committed 331 days ago

Verified 92873377

8740 add single processing to getting started instructions (#8875)

pgmoka committed 332 days ago

Verified e341ff09

[ragged-paged-attn] Combine k_pages and v_pages on num_kv_head (#8892)

bythew3i committed 332 days ago

Verified 7a3c0518

Encapsulate Mesh invariants (#8882)

rpsilva-aws committed 332 days ago

Verified 96ad8f5f

Avoid unnecessary copy in TensorSource (#8849)

lsy323 committed 332 days ago

Verified 8dc5b496

update Dockerfile config for CI (#8886)

zpcore committed 333 days ago

Verified 3066b034

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

tengyifei committed 333 days ago

Verified 6a7ce475

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

tengyifei committed 334 days ago

Verified a3ef52e5

Add lowering for `bitwise_right_shift`. (#8866)

ysiraichi committed 335 days ago

Verified 2aa11ccc

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

iwknow committed 335 days ago

Verified f2bdecfa

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

pgmoka committed 335 days ago

Verified 3808df01

Add lowering for `bitwise_left_shift`. (#8865)

ysiraichi committed 335 days ago

Verified 46a7990f

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

bhavya01 committed 337 days ago

Verified 8030f059

Support collective matmul optimization in mp (#8855)

yaochengji committed 338 days ago

Verified 2ad5b5d3

fix CPP test build failure in upstream (#8867)

zpcore committed 338 days ago

Verified 760675ad

Avoid destorying the cluster (#8864)

tengyifei committed 339 days ago

Verified c6e55ba5

Fix missing min_tpu_nodes argument (#8863)

tengyifei committed 339 days ago

Verified ece5fe7e

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

tengyifei committed 339 days ago

Verified 6a93dcfb

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

bythew3i committed 340 days ago

Verified 4190fc0e

fix dependency (#8852)

zpcore committed 340 days ago

Verified c7d0b1ef

fix existing bazel link removal (#8848)

zpcore committed 341 days ago

Verified 18cdb17a

Make cxx_abi enable by default (#8844)

zpcore committed 341 days ago

Verified 8f314a5a

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

zpcore committed 341 days ago

Verified 4c5cc420

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

pgmoka committed 342 days ago

Verified 76b0ce50

Newer Older