pytorch/xla

Pull Requests Commits

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

[torch_xla2] summary hard ops to non_support_ops_list (#8235)

ManfeiBai committed 1 year ago

Verified 07d0823f

[torch_xla2] add some ops (#8228)

ManfeiBai committed 1 year ago

Verified 93b4a90e

Add diagonal_scatter aten ops (#8232)

nupurbaghel committed 1 year ago

Verified c29cccf7

Add support for linalg.lu (#8227)

matinehAkhlaghinia committed 1 year ago

Verified 7aa996cf

[torch_xla2] Fix geometric and gcd (#8226)

dvhg committed 1 year ago

Verified d50fecb0

Raise error when minibatch is used in SPMD dataloading and per host b… (#8205)

JackCaoG committed 1 year ago

Verified e3cf356a

Update trigger to 2.5.0-rc9 for 2.5 release (#8221)

ManfeiBai committed 1 year ago

Verified 14d14aec

[torch_xla2] Fix cholesky_inverse (#8214)

dvhg committed 1 year ago

Verified eba28f17

Add nextafter op (#8218)

tracych477 committed 1 year ago

Verified 922805e4

reimplement jax.image.scale_and_translate instead of jax.image.resize for _upsample_bilinear2d_aa (#8208)

barney-s committed 1 year ago

Verified d3ff030b

Newer Older