pytorch/xla

Pull Requests Commits

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

dlibenzi committed 6 years ago

575e9237

Fix copy_() operation.

dlibenzi committed 6 years ago

0210cde0

Remove unused header.

Ailing Zhang committed 6 years ago

9e05cefa

Use more XLA constants.

dlibenzi committed 6 years ago

c1b6ee90

Fix deps of googletest before the broken commit.

Ailing Zhang committed 6 years ago

ac897428

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

mruberry committed 6 years ago

Verified 2c14f650

Skip test_softshrink_negative

xuhdev committed 6 years ago

edafa973

Bring in new TF.

dlibenzi committed 6 years ago

b68a0422

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

dlibenzi committed 6 years ago

9a32920c

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

mruberry committed 6 years ago

0011ce87

Skip two new tests. (#1620)

ailzhang committed 6 years ago

66cd32a6

Allow proper inplace copy on different devices.

dlibenzi committed 6 years ago

069cae8e

Fix XRT metric remap name. (#1614)

dlibenzi committed 6 years ago

2971548d

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

mruberry committed 6 years ago

f4da6c42

Fix typo in reporting aten_ops_sum to Tensorboard (#1612)

zcain117 committed 6 years ago

49265e9d

Restores test order

mruberry committed 6 years ago

991dd3f2

Adds TestViewOps to test suite

Mike Ruberry committed 6 years ago

6c480428

Write an initial event to Tensorboard when tests start up. This allow… (#1610)

zcain117 committed 6 years ago

d06c9f19

Fix GCSFS docs.

dlibenzi committed 6 years ago

72c22c78

Bring back split*() since pytorch PR got reverted.

dlibenzi committed 6 years ago

455a1461

Use C++ interface for GCSFS.

dlibenzi committed 6 years ago

0710b87f

Use bazelisk instead of specifying bazel version manually. (#1601)

ailzhang committed 6 years ago

Verified 5ff4a300

Expose TF files. (#1606)

dlibenzi committed 6 years ago

Verified 7525565c

Add simple GCSFS benchmark.

dlibenzi committed 6 years ago

608cc8d7

Use upper layer API for metrics.

dlibenzi committed 6 years ago

ea1b01bc

Print metrics report only from master device. (#1597)

taylanbil committed 6 years ago

Verified 84c93754

Rethrow the exception within the async task. (#1603)

dlibenzi committed 6 years ago

Verified 96efb8bd

Measure the tensor transform time separately.

dlibenzi committed 6 years ago

ed134aba

Rebase on top master before running CI build. (#1600)

ailzhang committed 6 years ago

Verified fdf2f69c

Remove override for randperm (#1598)

ailzhang committed 6 years ago

Verified 237d3262

Newer Older