pytorch/xla

Pull Requests Commits

Update jaten.py

ManfeiBai committed 1 year ago

Verified fd8715af

[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

add dist op doc (#8273)

zpcore committed 1 year ago

Verified fa311ec8

Add peak_bytes_used to the get_memory_info (#8265)

JackCaoG committed 1 year ago

Verified a0f81e5e

Add support for linalg.lstsq (#8261)

matinehAkhlaghinia committed 1 year ago

Verified 32afdbbb

Fix op info test for linalg.lu_factor and linalg.ldl_factor (#8263)

dvhg committed 1 year ago

Verified 349958fb

Add `unique`, `unique_consecutive` (#8258)

mrguenther committed 1 year ago

Verified ee388f65

Update artifacts.auto.tfvars for 2.5.0-rc10 (#8264)

ManfeiBai committed 1 year ago

Verified c895c5a2

Set weights_only to False for xser.load (#8259)

JackCaoG committed 1 year ago

Verified f71c02d1

Avoid running tpu_info cli test on Trillium (#8257)

bhavya01 committed 1 year ago

Verified ecdc68bb

Fix nn.functional.bilinear (#7517) (#8256)

simonteozw committed 1 year ago

Verified 7760d87b

Add _aten_cdist Op (#8238)

vyom1611 committed 1 year ago

Verified 80db07b2

Older