diff options
author | 2017-12-13 13:45:05 -0800 | |
---|---|---|
committer | 2017-12-13 13:48:42 -0800 | |
commit | 70062d11bf11d6579bfdbc87c3350a0074a12ae8 (patch) | |
tree | 3be96fe513d9273c08b1bf489d6178e849d69918 /tensorflow/contrib/nccl | |
parent | 553e8f14c8c025a8c09e0a6cb824c786bc258f56 (diff) |
Rename Stream::BlockHostUntilDoneWithStatus to BlockHostUntilDone.
PiperOrigin-RevId: 178951330
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r-- | tensorflow/contrib/nccl/kernels/nccl_manager_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc b/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc index ef76df6b0d..658baf18d3 100644 --- a/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc +++ b/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc @@ -175,7 +175,7 @@ class NcclManagerTest : public ::testing::Test { auto out_gpu_mem = AsDeviceMemory(out_gpu.flat<float>().data()); stream->ThenMemcpy(out_cpu.flat<float>().data(), out_gpu_mem, out_cpu.TotalBytes()); - SE_ASSERT_OK(stream->BlockHostUntilDoneWithStatus()); + SE_ASSERT_OK(stream->BlockHostUntilDone()); test::ExpectTensorEqual<float>(test_case->expected, out_cpu); } } @@ -236,7 +236,7 @@ TEST_F(NcclManagerTest, MultipleCallers) { for (int i = 0; i < num_ranks; ++i) { auto* device = devices->at(i % devices->size()); auto* stream = device->tensorflow_gpu_device_info()->stream; - SE_ASSERT_OK(stream->BlockHostUntilDoneWithStatus()); + SE_ASSERT_OK(stream->BlockHostUntilDone()); } std::random_shuffle(case_and_device_num.begin(), case_and_device_num.end()); |