pytorch/xla

Pull Requests Commits

Fix the test by avoiding the current (will be fixed soon) XLA AllReduce bug on tuple inputs with single element inside.

dlibenzi committed 5 years ago

b79fac57

Fix MP AllReduce test.

dlibenzi committed 5 years ago

93f6d209

Do not spawn a different process if we will be running on one single core.

dlibenzi committed 5 years ago

04944651

Allow pow() to take int exponents as the lowering does the promotions already.

dlibenzi committed 5 years ago

6e436eb4

Temporarily disable op-by-op test.

dlibenzi committed 5 years ago

0bd9a898

Use shape_fn for expensive shape calculations into GenericOp(). (#1410)

dlibenzi committed 5 years ago

954c99a2

Added aten::reflection_pad2d and aten::reflection_pad2d_backwar… (#1409)

dlibenzi committed 5 years ago

012367f6

Propagate requires_grad when sending to XLA device. (#1411)

dlibenzi committed 5 years ago

b618a917

Don't use auto in test

vishwakftw committed 5 years ago

c873a6dd

Remove logdet.h/logdet.cpp, and migrate implementation to ops.h/ops.cpp

vishwakftw committed 5 years ago

ecac1565

Fix build by making node

vishwakftw committed 5 years ago

9025c22a

Add aten::logdet op

vishwakftw committed 5 years ago

70c1ceaa

Migrate code to use the new XlaBuilder::GetShapePtr().

dlibenzi committed 5 years ago

b255f1d1

add hypothesis to conda install packages

zcain117 committed 5 years ago

078bd7b8

Bring new TF with GetShapePtr() changes.

dlibenzi committed 5 years ago

8ffcb772

Implement aten::nonzero and aten::masked_select for TPU.

dlibenzi committed 5 years ago

4f923116

Disable test test_min_max_binary_op_nan (#1397)

xuhdev committed 5 years ago

51f0c87c

Use Flatten() in other places to avoid creating noop Reshape()s.

dlibenzi committed 5 years ago

7a6f460c

Cast back result to original type if both inputs had teh same b… (#1396)

dlibenzi committed 5 years ago

6bf837d2

Added support for aten::take and its backward (aten::put_).

dlibenzi committed 5 years ago

b7ca463e

disable some tests from the batch now able to run

zcain117 committed 5 years ago

bee45f97

Skips test_pred_and_u8 (#1392)

mruberry committed 5 years ago

Verified a714d8c0

Update missed notebook with version selection

jysohn23 committed 5 years ago

b91360ec

Update notebooks with version changing capability (#1387)

jysohn23 committed 5 years ago

Verified 32cc4232

Redo serialization patch (#1388)

ailzhang committed 5 years ago

Verified 4798bc40

Collect torchvision wheel artifact (#1386)

jysohn23 committed 5 years ago

Verified 463b8c69

Runs type promotion tests (#1384)

mruberry committed 5 years ago

Verified d3096b1e

Disable python operations tests that fail on TPU (#1382)

zcain117 committed 5 years ago

Verified 70e754b6

use master branch of torchvision for pytorch/xla docker images (#1383)

zcain117 committed 5 years ago

Verified ccc1c6ff

Added utility API to optimize the events which require tensor data materialization.

dlibenzi committed 5 years ago

0303beeb

Older