diff options
author | Mark Heffernan <meheff@google.com> | 2018-05-01 20:14:29 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-01 20:17:23 -0700 |
commit | 7715b7b0650c2f20b47189a060580a45e510acd8 (patch) | |
tree | 9bdec673ce69795e34b2f3f520b64214f08471f7 /tensorflow/compiler/xla/service/buffer_assignment_test.cc | |
parent | b50f6325143486eb82b5654f8794f0771b54dd4d (diff) |
Add missing colocated element in test in buffer_assignment_test.
This was resulting in a flaky test because sometimes the live set would include this missing colocated element perhaps because the buffers in the allocation has some nondeterministic order (read from a map?).
PiperOrigin-RevId: 195039311
Diffstat (limited to 'tensorflow/compiler/xla/service/buffer_assignment_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/buffer_assignment_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/buffer_assignment_test.cc b/tensorflow/compiler/xla/service/buffer_assignment_test.cc index 40cf6483aa..f6d6b5c36a 100644 --- a/tensorflow/compiler/xla/service/buffer_assignment_test.cc +++ b/tensorflow/compiler/xla/service/buffer_assignment_test.cc @@ -1634,6 +1634,7 @@ TEST_F(BufferAssignmentTest, PeakBuffersWhile) { } EXPECT_EQ(bcast_buffer->instruction(), bcast); EXPECT_TRUE( + nonbcast_buffer->instruction() == copy || nonbcast_buffer->instruction() == while_op || nonbcast_buffer->instruction() == body->parameter_instruction(0) || nonbcast_buffer->instruction() == body->root_instruction() || |