pytorch/xla

Pull Requests Commits

Added simple thread pool instead of spawning new threads.

dlibenzi committed 7 years ago

2ef3227c

Fix the hanging issue on multi-core.

dlibenzi committed 7 years ago

92c08e06

MultiCore API tweaks and adding back removed models, now using the MultiCore API.

dlibenzi committed 7 years ago

ad330b33

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

dlibenzi committed 7 years ago

1f184b3a

Refactor DataParallel to support generic input types.

dlibenzi committed 7 years ago

2b16faf1

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

dlibenzi committed 7 years ago

f0421341

Fix MNIST to set the default device.

dlibenzi committed 7 years ago

8122f09d

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

dlibenzi committed 7 years ago

b6c448b6

Fixed the op-by-op parameter type.

dlibenzi committed 7 years ago

5b802fab

Better promotion for PREDs.

dlibenzi committed 7 years ago

35e06826

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

dlibenzi committed 7 years ago

9c7860d2

Use the new XRTExecuteChained operation.

dlibenzi committed 7 years ago

25618c4a

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

dlibenzi committed 7 years ago

ce69bd5c

Allow the PyTorch include folder to be overridden via environment.

dlibenzi committed 7 years ago

4404ef44

Add gradient clipping patch

Alex Şuhan committed 7 years ago

8fcdcb39

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

dlibenzi committed 7 years ago

d91a8acc

Remove more JIT leftovers

asuhan committed 7 years ago

28dcfbef

Remove autodiff patch as well

asuhan committed 7 years ago

3ce4e992

Remove the JIT code and all its dependencies.

dlibenzi committed 7 years ago

b7c0ab3f

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

dlibenzi committed 7 years ago

c96789ef

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

Alex Şuhan committed 7 years ago

4f99844f

Remove patch which landed

Alex Şuhan committed 7 years ago

262c360b

Use two-pass lowerings for argmin and argmax

Alex Şuhan committed 7 years ago

c9d654ad

Fix pending PR extension

Alex Şuhan committed 7 years ago

a5d996d0

Fix XlaModule constructor after PyTorch changes

Alex Şuhan committed 7 years ago

470b15e1

Local fix for max_pool2d differentiation

Alex Şuhan committed 7 years ago

160d9cb5

Use xla::TorchGather to lower Gather

Alex Şuhan committed 7 years ago

eb769a38

Use xla::TorchIndexSelect to lower IndexSelect

Alex Şuhan committed 7 years ago

1080e7f7

Add pending PyTorch patch fixing equal for boolean tensors

asuhan committed 7 years ago

bde41239

Update to latest TF

Alex Şuhan committed 7 years ago

73919342

Older