pytorch/xla

Pull Requests Commits

Avoid copying TF string into std:string.

dlibenzi committed 5 years ago

193ecc38

Add test for rendezvous to the set of python tests.

dlibenzi committed 5 years ago

ec3336fa

Disable test_eig_with_eigvec which fails for precision on XLA.

dlibenzi committed 5 years ago

c0205fa5

Added user visible rendezvous interface.

dlibenzi committed 5 years ago

97f461f0

Make reductions work with dynamic shapes.

dlibenzi committed 5 years ago

b1eeb64a

Revert "Fix copy_ from xla -> cpu."

dlibenzi committed 5 years ago

6cad3b18

Disable digamma test failing on TPU.

dlibenzi committed 5 years ago

b06662c8

Skip test OOMing.

dlibenzi committed 5 years ago

a4155ce2

Add logsumexp op

JackCaoG committed 5 years ago

49c719ec

Fix bitwise ops shape inference.

dlibenzi committed 5 years ago

e2c1d5ef

Emit the tensors graph before compiling to make sure some level of debugging can be done on the TEXT graph in case of XLA lowering errors.

dlibenzi committed 5 years ago

08e4ae12

Correctly marshal the new XRT metrics.

dlibenzi committed 5 years ago

3109a2dd

Partitions the transfer to server to avoid tripping the Protocol Buffers 2GB limit.

dlibenzi committed 5 years ago

01dd6f96

Fix copy_ from xla -> cpu.

Ailing Zhang committed 5 years ago

8b389f67

Revert "Keep the memory used for sending tensors to device under control, to avoid tripping over the ProtocolBuffer 2GB time bomb."

dlibenzi committed 5 years ago

f77a9474

Keep the memory used for sending tensors to device under control, to avoid tripping over the ProtocolBuffer 2GB time bomb.

dlibenzi committed 5 years ago

575e9237

Fix copy_() operation.

dlibenzi committed 5 years ago

0210cde0

Remove unused header.

Ailing Zhang committed 5 years ago

9e05cefa

Use more XLA constants.

dlibenzi committed 5 years ago

c1b6ee90

Fix deps of googletest before the broken commit.

Ailing Zhang committed 5 years ago

ac897428

Adds multicore AlexNet on Fashion MNIST to list of README Colabs (#1637)

mruberry committed 5 years ago

Verified 2c14f650

Skip test_softshrink_negative

xuhdev committed 5 years ago

edafa973

Bring in new TF.

dlibenzi committed 5 years ago

b68a0422

Preparing for new metrics coming in from TPU software, and cleanup include barrier.

dlibenzi committed 5 years ago

9a32920c

Adds multicore alexnet trained on fashion mnist colab (#1621)

mruberry committed 5 years ago

0011ce87

Skip two new tests. (#1620)

ailzhang committed 5 years ago

66cd32a6

Allow proper inplace copy on different devices.

dlibenzi committed 5 years ago

069cae8e

Fix XRT metric remap name. (#1614)

dlibenzi committed 5 years ago

2971548d

Updates AlexNet on Fashion MNIST + adds README link (#1613)

mruberry committed 5 years ago

f4da6c42

Fix typo in reporting aten_ops_sum to Tensorboard (#1612)

zcain117 committed 5 years ago

49265e9d

Older