pytorch/xla

Pull Requests Commits

Update test_ops.py

ManfeiBai committed 1 year ago

Verified 6572c0a4

Update test_ops.py

ManfeiBai committed 1 year ago

Verified c2a5c344

Update jaten.py

ManfeiBai committed 1 year ago

Verified 56c7a52e

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

Add new_empty_strided op (#8217)

tracych477 committed 1 year ago

Verified 02d90472

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

rpsilva-aws committed 1 year ago

Verified 0b19374c

Update decomp table (#8210)

tugsbayasgalan committed 1 year ago

Verified a2681e86

[export] Conditionally load decomp table (#8209)

lsy323 committed 1 year ago

Verified c22dfe98

Fix warning messages when using fsdpv2 (#8207)

JackCaoG committed 1 year ago

Verified 48137028

Allow MpDeviceLoader to shard dictionaries of tensor (#8202)

bhavya01 committed 1 year ago

Verified cc631b9b

Update actions runner version for TPU CI (#8206)

mbzomowski committed 1 year ago

Verified b9e65e74

First step in implementing upsample_bilinear2d_aa (#8090)

barney-s committed 1 year ago

Verified 365841da

[benchmarks] Default to functionalization disabled. (#8093)

ysiraichi committed 1 year ago

Verified b6bcf03f

Fix lgamma, mvlgamma, multinomial, and nanmedian (#7513) (#8095)

simonteozw committed 1 year ago

Verified 5dbdb8db

Enable cross entropy loss for xla autocast with FP32 precision (#7992) (#8094)

avizon-aws committed 1 year ago

Verified 940bee45

Fix torch.min and torch.mode (#7513) (#8092)

simonteozw committed 1 year ago

Verified 0f645b27

Remove linalg.matrix_rank from skiplist (#7497) (#8096)

simonteozw committed 1 year ago

Verified d98be5b9

Implement `torch.ops.aten.embedding_renorm_` (#8091)

guyao committed 1 year ago

Verified 5872b209

Add new_empty (#8087)

tracych477 committed 1 year ago

Verified ecc0f5ab

Update README.md to 2.6.0.dev for nightly wheel (#8080)

ManfeiBai committed 1 year ago

Verified 248a5bd5

Reduced the max_unpoolxd logic into one function #7524 (#8085)

hosseinsarshar committed 1 year ago

Verified 45d0e222

Added the support for max_unpool1d, max_unpool2d, and max_unpool3d #7524 (#8084)

hosseinsarshar committed 1 year ago

Verified 5f3c983e

Update trigger to 2.5.0-rc5 for 2.5 release (#8083)

ManfeiBai committed 1 year ago

Verified 597bb296

Older