pytorch/xla

Pull Requests Commits

Use exceptions directly to propagate errors, instead of channeling to Status.

dlibenzi committed 6 years ago

1f184b3a

Refactor DataParallel to support generic input types.

dlibenzi committed 6 years ago

2b16faf1

Removed no more used code in tensor.* due to JIT dropping.

dlibenzi committed 6 years ago

f0421341

Fix MNIST to set the default device.

dlibenzi committed 6 years ago

8122f09d

Introduce an environment flag to avoid building C++ tests (used for Docker).

dlibenzi committed 6 years ago

b6c448b6

Fixed the op-by-op parameter type.

dlibenzi committed 6 years ago

5b802fab

Better promotion for PREDs.

dlibenzi committed 6 years ago

35e06826

Drop the device data now that we are no more using JIT.

dlibenzi committed 6 years ago

9c7860d2

Use the new XRTExecuteChained operation.

dlibenzi committed 6 years ago

25618c4a

Bring new TF and fix convolution breakage due to TF2XLA changes.

dlibenzi committed 6 years ago

ce69bd5c

Allow the PyTorch include folder to be overridden via environment.

dlibenzi committed 6 years ago

4404ef44

Add gradient clipping patch

asuhan committed 6 years ago

8fcdcb39

Track post order size in op-by-op mode.

dlibenzi committed 6 years ago

d91a8acc

Remove more JIT leftovers

asuhan committed 6 years ago

28dcfbef

Remove autodiff patch as well

asuhan committed 6 years ago

3ce4e992

Remove the JIT code and all its dependencies.

dlibenzi committed 6 years ago

b7c0ab3f

Added split executor capability, which runs a given IR graph one IR node at a time.

dlibenzi committed 6 years ago

c96789ef

Don't use torchscript version for aten::max_pool2d autodiff

asuhan committed 6 years ago

4f99844f

Remove patch which landed

asuhan committed 6 years ago

262c360b

Use two-pass lowerings for argmin and argmax

asuhan committed 6 years ago

c9d654ad

Fix pending PR extension

asuhan committed 6 years ago

a5d996d0

Fix XlaModule constructor after PyTorch changes

asuhan committed 6 years ago

470b15e1

Local fix for max_pool2d differentiation

asuhan committed 6 years ago

160d9cb5

Use xla::TorchGather to lower Gather

asuhan committed 6 years ago

eb769a38

Use xla::TorchIndexSelect to lower IndexSelect

asuhan committed 6 years ago

1080e7f7

Add pending PyTorch patch fixing equal for boolean tensors

asuhan committed 6 years ago

bde41239

Update to latest TF

asuhan committed 6 years ago

73919342

Add step-mark operation, and split sync-tensors into a separate operation.

dlibenzi committed 6 years ago

9e59aa5c

Simplify XLA tensor code generator further, after the recent ATEN changes.

dlibenzi committed 6 years ago

de703fa7

Fix XLA tensor code generation to cope with the PyTorch/ATEN changes.

dlibenzi committed 6 years ago

2ba3696c

Newer Older