pytorch/xla

Pull Requests Commits

Explicitly use python3 for test/run_tests.sh

jysohn23 committed 6 years ago

c98ec35e

Limit the check graph frequency as well, until the XLA bug has been fixed.

dlibenzi committed 6 years ago

769109e4

Use common syn function for apply tensors graph.

dlibenzi committed 6 years ago

0a99322c

Add support for _th_le, _th_ne

asuhan committed 6 years ago

72caf51f

Send more tensors in parallel to allow the parallelism used by the XRT computation client to reduce the send time.

dlibenzi committed 6 years ago

1723eb55

Cache the Tensorflow tensor memory allocations to reduce clear_page_c performance hits.

dlibenzi committed 6 years ago

2360dae0

Remove more unused code in test/test_operations.py

asuhan committed 6 years ago

d8899009

Added simple thread pool instead of spawning new threads.

dlibenzi committed 6 years ago

d581df31

Fix scalar type of sum

asuhan committed 6 years ago

8885ba67

Fix the hanging issue on multi-core.

dlibenzi committed 6 years ago

92c08e06

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

dlibenzi committed 6 years ago

ad330b33

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

Newer Older