aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/nccl
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-01-09 11:46:32 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-09 11:53:04 -0800
commit3d00ca56b0a85083ab9b51a325ba11c9ebbe7890 (patch)
tree717f9088770f3b6249efaf583307c0efacadc9be /tensorflow/contrib/nccl
parent0a700fb22e79f493ff75d910f49162f713ee9c4e (diff)
Replaces std::random_shuffle by std::shuffle for C++17 compatibility.
PiperOrigin-RevId: 181350723
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r--tensorflow/contrib/nccl/kernels/nccl_manager_test.cc4
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc b/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc
index 658baf18d3..985b2bae25 100644
--- a/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc
+++ b/tensorflow/contrib/nccl/kernels/nccl_manager_test.cc
@@ -16,6 +16,7 @@ limitations under the License.
#ifdef GOOGLE_CUDA
#include <algorithm>
+#include <random>
#include <vector>
#include "tensorflow/contrib/nccl/kernels/nccl_manager.h"
@@ -239,7 +240,8 @@ TEST_F(NcclManagerTest, MultipleCallers) {
SE_ASSERT_OK(stream->BlockHostUntilDone());
}
- std::random_shuffle(case_and_device_num.begin(), case_and_device_num.end());
+ std::shuffle(case_and_device_num.begin(), case_and_device_num.end(),
+ std::mt19937(std::random_device()()));
mutex mu; // guards case_and_device_num.
std::unique_ptr<thread::ThreadPool> pool(