pytorch/xla

Pull Requests Commits

Add support for IR graph cloning.

dlibenzi committed 6 years ago

d25897f0

Fixed default img dimension bug

taylanbil committed 6 years ago

db2903ad

google-pyformat

taylanbil committed 6 years ago

0165998b

bug fix add missing '

taylanbil committed 6 years ago

85a3d763

Added model properties dict to handle special models

taylanbil committed 6 years ago

59243511

Adding (almost) all torchvision models to imgnet

taylanbil committed 6 years ago

cae1bf08

Install clang-7 from LLVM directly (#740)

jysohn23 committed 6 years ago

Verified 4df0e904

Fix torch.save workaround

asuhan committed 6 years ago

6b66544c

Add _index_put_impl_ to ATen XLA tensor

asuhan committed 6 years ago

d1cebc64

Fix ELU implementation

asuhan committed 6 years ago

86281b63

Collect wheels from nightly image (#735)

jysohn23 committed 6 years ago

Verified 354aa529

added swap and pyc files to gitignore

taylanbil committed 6 years ago

758f8bd2

Code formatting

taylanbil committed 6 years ago

d08c5e4f

Added model_specific defaults to params such as num_epochs, lr etc.

taylanbil committed 6 years ago

fcb61d9d

add a --model flag to use any (supported) torchvision model

taylanbil committed 6 years ago

c7b87dce

Dump the metrics report if the environment variable METRICS_DEBUG is set.

dlibenzi committed 6 years ago

a76e3c65

Added GRPC service to coordinate hosts in a client mesh.

dlibenzi committed 6 years ago

8d67a32b

Ignore layout when computing strides

asuhan committed 6 years ago

0e69fd5c

Add the initial support for local vs. remote devices, to support feeding slices bigger than single TPU host.

dlibenzi committed 6 years ago

7008ac11

Fix u8 dtype cancellation.

dlibenzi committed 6 years ago

21d16c2e

Block deployment on CPU smoke tests (#725)

jysohn23 committed 6 years ago

Verified 697c2adb

Update README with docker distribution (#723)

jysohn23 committed 6 years ago

Verified 882505a0

Allow saving graphs in HLO format as well.

dlibenzi committed 6 years ago

bf585125

Add API to retrieve XLA internal device names.

dlibenzi committed 6 years ago

eb5db322

Tag the docker builds (#721)

jysohn23 committed 6 years ago

Verified 9de32aec

Add local fix for torch.save on XLA tensors

asuhan committed 6 years ago

a3a246d3

Choose optimal TPU layout for all ranks for which makes sense.

dlibenzi committed 6 years ago

f1ba1cf1

Avoid routing special scalar to device data, to give a chance to the XLA compiler to simplify code.

dlibenzi committed 6 years ago

0a1227b5

Re-align code after PT Variable code changes.

dlibenzi committed 6 years ago

98d669ee

Dockerize Builds (#712)

jysohn23 committed 6 years ago

Verified 918e1ce6

Newer Older