pytorch/xla

Pull Requests Commits

Add benchmark with transfer in loop.

dlibenzi committed 7 years ago

524829c5

More concise error message in CheckIntList

Alex Şuhan committed 7 years ago

88736aa2

Clean up lambda captures in pooling operations

Alex Şuhan committed 7 years ago

93956a9d

Resize kernel, stride and padding for pooling operations

Alex Şuhan committed 7 years ago

5c498c90

Support for n-dimensional max pooling

Alex Şuhan committed 7 years ago

13bbaa06

Support for n-dimensional average pooling

Alex Şuhan committed 7 years ago

862aceb7

Merge pull request #532 from pytorch/add_py_bench

dlibenzi committed 7 years ago

Verified 2509d5e7

Added Python benchmark to measure Cloud TPU latency testing.

dlibenzi committed 7 years ago

02a01396

Fix the XLA device guard to add noexcept qualifier.

dlibenzi committed 7 years ago

9e7a8239

Fix Colab to use JIT MNIST as the XTen version is not yet supported in the wheels we can supply for TF 1.13 compatibility.

dlibenzi committed 7 years ago

3b8bda4c

Fixes zeros_like() monkey-patch in XLA Python init.

dlibenzi committed 7 years ago

2ffe5a69

Support direct remap ATEN generation to be used for TH functions (and similar other scenarios).

dlibenzi committed 7 years ago

b567bbdb

Count XLA tensors.

dlibenzi committed 7 years ago

c79202dd

Bump compilation caches up and made them environment configurable.

dlibenzi committed 7 years ago

33358305

Fix selected slice shape.

dlibenzi committed 7 years ago

08f8d6f8

Bound the step for slices.

dlibenzi committed 7 years ago

0e9c6f6c

Remove TensorData and use DeviceData back as it is no more needed.

dlibenzi committed 7 years ago

a4b7006e

Added Select and Unselect operations to handle slices with strides != 1.

dlibenzi committed 7 years ago

eb316286

Enable the model comparator to be left active and unconfigured, resulting in noop.

dlibenzi committed 7 years ago

470c9444

Added aten::triangular_solve op

vishwakftw committed 7 years ago

3e5051fb

Fix XLA tensor views which are not handled by the XLA tensor core.

dlibenzi committed 7 years ago

3ff502ba

Minor cleanup of model comparator.

dlibenzi committed 7 years ago

aae5e5cf

Added model comparator Python module.

dlibenzi committed 7 years ago

1dfe3e2b

Fix aten::clamp to use lowest() instead of min().

dlibenzi committed 7 years ago

7746ff65

Replace IntList with IntArrayRef

Alex Şuhan committed 7 years ago

90849b0f

Add index_{add, copy} and their inplace versions to ATen XLA tensor

Alex Şuhan committed 7 years ago

78a09d93

No need to add CRS for one device replications.

dlibenzi committed 7 years ago

963e6fa6

Added script to normalize the IR graph text output, to allow textual diffs.

dlibenzi committed 7 years ago

5686da1d

Enable shape caching for IndexOp

Alex Şuhan committed 7 years ago

d8be8bd9

Replace Shape::dimensions_size() uses with Shape::rank()

Alex Şuhan committed 7 years ago

717db87f

Older