pytorch/xla

Pull Requests Commits

Simplify APIs and remove redundant logging

asuhan committed 6 years ago

0288df9a

Address more review feedback

asuhan committed 6 years ago

e0f13451

Fix the construction of argument list for backward

asuhan committed 6 years ago

e9248d3c

Even more cleanups from review feedback

asuhan committed 6 years ago

3be22fed

Additional XlaModule fixes per review feedback

asuhan committed 6 years ago

5f564015

Store gradient descriptor in XlaModule

asuhan committed 6 years ago

5cbc0aa4

Additional cleanups from PR feedback

asuhan committed 6 years ago

f5b5faa9

Adjust signature of RemoveUnusedForwardOutputs

asuhan committed 6 years ago

80a8da60

Address some of the logging PR feedback

asuhan committed 6 years ago

6b217690

Add and use XlaHelpers::SizesOfXlaOp helper

asuhan committed 6 years ago

02832684

Remove nearly all dependencies on PyTorch trace shapes

asuhan committed 6 years ago

842ca03b

Add pass to replace inplace operators with non-inplace counterparts

asuhan committed 6 years ago

e849a657

Add pass to remove in-place operations on output parameters

asuhan committed 6 years ago

4b1edad5

Keep track of evaluated aten::size

asuhan committed 6 years ago

7ded34a6

Initial changes for latest PyTorch

asuhan committed 6 years ago

abd38b35

Merge pull request #62 from pytorch/test_util_restructure1

jysohn23 committed 6 years ago

Verified d4fbf359

Do not import xla_model.py from test_util.py, as this ends up picking up the xla_model.py from the installed module, instead of the local one.

dlibenzi committed 6 years ago

b709d55e

Add tensorboardX support for all our models

jysohn23 committed 6 years ago

Verified 161686d9

Addressed review comments and ran google-pyformat

jysohn23 committed 6 years ago

cd35dc52

Handle cases for metrics_debug

jysohn23 committed 6 years ago

ea068be1

Remove explicit writer from cifar and imagenet models

jysohn23 committed 6 years ago

b3e9cfcf

Fix processed examples calculated

jysohn23 committed 6 years ago

7ec1597c

Create new classes for writing and logging metrics

jysohn23 committed 6 years ago

1d410082

Merge branch 'jysohn/tbx' of https://github.com/pytorch/xla into jysohn/tbx

jysohn23 committed 6 years ago

016b6aa2

Add tensorboardX support for convergence tests

jysohn23 committed 6 years ago

f6c85c9b

Add tensorboardX support for test_train_mnist.py

jysohn23 committed 6 years ago

4f33a5c2

Dedicate IO thread pool to non CPU bound closures.

dlibenzi committed 6 years ago

fa355a71

Make execute parallel run the multiple sessions in parallel.

dlibenzi committed 6 years ago

e40f3117

Use the number of devices as default value for the number of releaser threads.

dlibenzi committed 6 years ago

b5b308c5

Allow debug metrics to be dumped within the train loop.

dlibenzi committed 6 years ago

ca97e619

Newer Older