pytorch/xla

Pull Requests Commits

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

Add support for erfinv (#8081)

matinehAkhlaghinia committed 1 year ago

Verified ed683817

Add stablediffusion inference reference model (#8027)

qihqi committed 1 year ago

Verified f088810a

Removed linalg.cond from skiplist (#8075)

anishfish2 committed 1 year ago

Verified f4c8b734

Removes linalg.eigvalsh from skiplist and fixes errors that occur using custom atol/rtol. (#8076)

anishfish2 committed 1 year ago

Verified 686f9080

Fix nprocs description for xmp.spawn; error if nprocs != 1 or None (#7971)

jeffhataws committed 1 year ago

Verified 3cd46778

Support dist.all_to_all_single (#8064)

zpcore committed 1 year ago

Verified b378a28c

Newer Older