pytorch/xla

Pull Requests Commits

Add profiler annotation for the decoderonly example

JackCaoG committed 1 year ago

852aa4fc

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

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

anishfish2 committed 1 year ago

Verified 855ec40e

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

tengyifei committed 1 year ago

Verified 4e691dd7

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

anishfish2 committed 1 year ago

Verified 72fda76e

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

dvhg committed 1 year ago

Verified 5b6e8cf1

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

ManfeiBai committed 1 year ago

Verified 22cf1976

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

ManfeiBai committed 1 year ago

Verified 47d231f3

Update tests for Neuron (#8053)

jeffhataws committed 1 year ago

Verified e657c872

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

will-cromar committed 1 year ago

Verified ea8c47a3

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

dvhg committed 1 year ago

Verified 6af550aa

[torch_xla2] Fix scatter and scatter_add (#8045)

dvhg committed 1 year ago

Verified 2aea97a7

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

dudulightricks committed 1 year ago

Verified d79a37c4

Update and short circuit the Neuron initialization (#8041)

rpsilva-aws committed 1 year ago

Verified d0ea5cc2

Op info test for linalg.norm, linalg.pinv, linalg.solve (#7503) (#8044)

yenkwang committed 1 year ago

Verified 441098b7

Fix minor grammar error (#8043)

simonteozw committed 1 year ago

Verified c0501f01

Fixed masked_scatter and masked_select (#8037)

anishfish2 committed 1 year ago

Verified 52673541

Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378) (#8038)

simonteozw committed 1 year ago

Verified b79a46b0

Fix quantile, nanquantile and gather (#8040)

matinehAkhlaghinia committed 1 year ago

Verified 6b33f8f6

Older