pytorch/xla

Pull Requests Commits

Update artifacts.auto.tfvars for rc10

ManfeiBai committed 342 days ago

Verified 4f61ea64

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

JackCaoG committed 342 days ago

Verified f71c02d1

Avoid running tpu_info cli test on Trillium (#8257)

bhavya01 committed 343 days ago

Verified ecdc68bb

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

simonteozw committed 344 days ago

Verified 7760d87b

Add _aten_cdist Op (#8238)

vyom1611 committed 344 days ago

Verified 80db07b2

enable lu, lu_solve testing (#8254)

barney-s committed 344 days ago

Verified 3635f6a2

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

simonteozw committed 344 days ago

Verified 82bcddad

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

dvhg committed 346 days ago

Verified a4e295a1

Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn (#8248)

anishfish2 committed 347 days ago

Verified f4f1386e

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

qihqi committed 347 days ago

Verified ab192b11

[torchxla2] add `argwhere` (#8073)

ManfeiBai committed 347 days ago

Verified 99c39108

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

dvhg committed 347 days ago

Verified ce7856fe

enable linspace & logspace tests (#8236)

barney-s committed 347 days ago

Verified 5a5165fd

enable linalg.vector_norm tests (#8249)

barney-s committed 347 days ago

Verified fe03cd23

Enabling log_normal tests (#8247)

barney-s committed 347 days ago

Verified e5e6d528

reorganized files and toc (#8246)

mikegre-google committed 347 days ago

Verified 543ccc86

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

yenkwang committed 347 days ago

Verified 0a91f799

add profiler hints in paralloader (#8244)

JackCaoG committed 348 days ago

Verified 8ec552d9

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

ManfeiBai committed 349 days ago

Verified 07d0823f

[torch_xla2] add some ops (#8228)

ManfeiBai committed 349 days ago

Verified 93b4a90e

Add diagonal_scatter aten ops (#8232)

nupurbaghel committed 350 days ago

Verified c29cccf7

Add support for linalg.lu (#8227)

matinehAkhlaghinia committed 350 days ago

Verified 7aa996cf

[torch_xla2] Fix geometric and gcd (#8226)

dvhg committed 350 days ago

Verified d50fecb0

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

JackCaoG committed 350 days ago

Verified e3cf356a

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

ManfeiBai committed 350 days ago

Verified 14d14aec

[torch_xla2] Fix cholesky_inverse (#8214)

dvhg committed 353 days ago

Verified eba28f17

Add nextafter op (#8218)

tracych477 committed 353 days ago

Verified 922805e4

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

barney-s committed 353 days ago

Verified d3ff030b

Add new_empty_strided op (#8217)

tracych477 committed 353 days ago

Verified 02d90472

Part 1: Introduce multi-node SPMD support for Neuron (#8204)

rpsilva-aws committed 354 days ago

Verified 0b19374c

Older