pytorch/xla

Pull Requests Commits

Update artifacts.auto.tfvars

ManfeiBai committed 1 year ago

Verified 2846164c

Convert entire model to jax xla (#8330)

barney-s committed 1 year ago

Verified 2592a68f

Fixed link to troubleshooting.md (#8321)

mikegre-google committed 1 year ago

Verified 3d1d1b9e

Add nightly C++11 ABI builds (#8307)

tengyifei committed 1 year ago

Verified e731688f

Add CODEOWNERS for /infra (#8325)

ManfeiBai committed 1 year ago

Verified 310ff8f4

[2.5.1 release] Update artifacts.auto.tfvars for 2.5.1 (#8320)

ManfeiBai committed 1 year ago

Verified 49ec0586

parallel_loader: fix AttributeError (#8314) (#8315)

mcuiaws committed 1 year ago

Verified 15aefe4d

[OpenXLA/libtpu pin Update] Update to Oct20 (#8311)

ManfeiBai committed 1 year ago

Verified 699daf27

[torch_xla2] Enable Background_Matting model by implement `aten::upsample_bilinear2d` (#8304)

ManfeiBai committed 1 year ago

Verified 2f5dc1f1

Add aten::linear op (#8302)

qihqi committed 1 year ago

Verified 5a4d7b4d

Fix torch.polar, torch.polygamma, torch.prod, torch.put (#7529) (#8303)

simonteozw committed 1 year ago

Verified d5eb4a07

[torch_xla2] add `igammac` (#8298)

ManfeiBai committed 1 year ago

Verified ff43125c

[torch_xla2] delete duplicated aten.trunc implementation (#8300)

ManfeiBai committed 1 year ago

Verified 25fc6303

[torch_xla2] Fix TF SavedModel export, add test (#8299)

GleasonK committed 1 year ago

Verified b2260a67

Add impl for torchvision.nms and support multiple batch dim for convolution. (#8291)

qihqi committed 1 year ago

Verified 39dc4bc0

Update test_ops.py with clean up (#8296)

ManfeiBai committed 1 year ago

Verified 7fd4e1cf

Update XLA pin to 10/16 (#8267)

JackCaoG committed 1 year ago

Verified 6536639e

Fix op info test for svd (#8289)

dvhg committed 1 year ago

Verified b8ee2c28

enable tensorsolve (#8287)

barney-s committed 1 year ago

Verified b2e0e33f

Add support for `torch.nn.functional.pad`. (#8290)

mrguenther committed 1 year ago

Verified e52dc4ac

disable GPU tests and build (#8286)

JackCaoG committed 1 year ago

Verified 387a274f

Add support for `torch.unravel_index`. (#8266)

mrguenther committed 1 year ago

Verified a3e4e12b

Update README.md fo 2.5 (#8281)

ManfeiBai committed 1 year ago

Verified d1a1966a

Fix torch digamma (#7383) (#8260)

simonteozw committed 1 year ago

Verified 6d690904

remove debug prints in lu_unpack implementation (#8282)

barney-s committed 1 year ago

Verified cec5052a

Fix support for `torch.var_mean`. (#8275)

mrguenther committed 1 year ago

Verified 2d73a5fd

Implement lu_unpack in jax (#8262)

barney-s committed 1 year ago

Verified fb34db8c

Update README.md for PyTorch/XLA 2.5 release (#8276)

ManfeiBai committed 1 year ago

Verified 74a02122

Update artifacts.auto.tfvars for 2.5.0 final wheel (#8272)

ManfeiBai committed 1 year ago

Verified 1bd17f7b

introduce torch_xla2.compile API, make sdxl to use it (#8269)

qihqi committed 1 year ago

Verified ddd4db78

Older