pytorch/xla

Pull Requests Commits

Update README.md

ManfeiBai committed 1 year ago

436e1a39

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

enable lu, lu_solve testing (#8254)

barney-s committed 1 year ago

Verified 3635f6a2

Implement avg_pool1d, avg_pool2d, and avg_pool3d (#7517) (#8255)

simonteozw committed 1 year ago

Verified 82bcddad

Fix op info test for linalg.inv and linalg.inv_ex (#8253)

dvhg committed 1 year ago

Verified a4e295a1

Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn (#8248)

anishfish2 committed 1 year ago

Verified f4f1386e

Revert "Fix op info test for linalg.inv and linalg.inv_ex (#8239)" (#8252)

qihqi committed 1 year ago

Verified ab192b11

[torchxla2] add `argwhere` (#8073)

ManfeiBai committed 1 year ago

Verified 99c39108

Fix op info test for linalg.inv and linalg.inv_ex (#8239)

dvhg committed 1 year ago

Verified ce7856fe

enable linspace & logspace tests (#8236)

barney-s committed 1 year ago

Verified 5a5165fd

enable linalg.vector_norm tests (#8249)

barney-s committed 1 year ago

Verified fe03cd23

Enabling log_normal tests (#8247)

barney-s committed 1 year ago

Verified e5e6d528

reorganized files and toc (#8246)

mikegre-google committed 1 year ago

Verified 543ccc86

Op info test for `linalg.solve_ex .. linalg.tensorinv` (#7504) (#8251)

yenkwang committed 1 year ago

Verified 0a91f799

add profiler hints in paralloader (#8244)

JackCaoG committed 1 year ago

Verified 8ec552d9

Older