pytorch/xla

Pull Requests Commits

Update actions runner version for TPU CI

mbzomowski committed 2 years ago

38d24db2

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

ysiraichi committed 2 years ago

Verified b6bcf03f

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

simonteozw committed 2 years ago

Verified 5dbdb8db

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

avizon-aws committed 2 years ago

Verified 940bee45

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

simonteozw committed 2 years ago

Verified 0f645b27

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

simonteozw committed 2 years ago

Verified d98be5b9

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

guyao committed 2 years ago

Verified 5872b209

Add new_empty (#8087)

tracych477 committed 2 years ago

Verified ecc0f5ab

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

ManfeiBai committed 2 years ago

Verified 248a5bd5

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

hosseinsarshar committed 2 years ago

Verified 45d0e222

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

hosseinsarshar committed 2 years ago

Verified 5f3c983e

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

ManfeiBai committed 2 years ago

Verified 597bb296

Add support for erfinv (#8081)

matinehAkhlaghinia committed 2 years ago

Verified ed683817

Add stablediffusion inference reference model (#8027)

qihqi committed 2 years ago

Verified f088810a

Removed linalg.cond from skiplist (#8075)

anishfish2 committed 2 years ago

Verified f4c8b734

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

anishfish2 committed 2 years ago

Verified 686f9080

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

jeffhataws committed 2 years ago

Verified 3cd46778

Support dist.all_to_all_single (#8064)

zpcore committed 2 years ago

Verified b378a28c

Fixed linalg.eigh by adding lowering and adding custom atol/rtol (#8072)

anishfish2 committed 2 years ago

Verified 855ec40e

[developer] Add a script to automate building torch_xla (#7876)

tengyifei committed 2 years ago

Verified 4e691dd7

Avoids error removing linalg.eig from skiplist by modifying atol and rtol (#8068)

anishfish2 committed 2 years ago

Verified 72fda76e

[torch_xla2] Fix op_info test for scatter_reduce (#8058)

dvhg committed 2 years ago

Verified 5b6e8cf1

Update trigger to 2.5.0-rc3 for 2.5 release (#8055)

ManfeiBai committed 2 years ago

Verified 22cf1976

add `resize_`, `resize_as_`, `rot90`, `repeat_interleave`, `combinations`, `diag_embed` (#8036)

ManfeiBai committed 2 years ago

Verified 47d231f3

Update tests for Neuron (#8053)

jeffhataws committed 2 years ago

Verified e657c872

[torch_xla2] Basic DDP example in a notebook (#8042)

will-cromar committed 2 years ago

Verified ea8c47a3

[torch_xla2] Fix nn.functional.conv2d and conv3d (#8048)

dvhg committed 2 years ago

Verified 6af550aa

[torch_xla2] Fix scatter and scatter_add (#8045)

dvhg committed 2 years ago

Verified 2aea97a7

parallel_loader: Fix TPU memory leak when calling __iter__. (#8039)

dudulightricks committed 2 years ago

Verified d79a37c4

Update and short circuit the Neuron initialization (#8041)

rpsilva-aws committed 2 years ago

Verified d0ea5cc2

Older