diff options
author | 2018-01-09 11:46:32 -0800 | |
---|---|---|
committer | 2018-01-09 11:53:04 -0800 | |
commit | 3d00ca56b0a85083ab9b51a325ba11c9ebbe7890 (patch) | |
tree | 717f9088770f3b6249efaf583307c0efacadc9be /tensorflow/contrib/nccl | |
parent | 0a700fb22e79f493ff75d910f49162f713ee9c4e (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.cc | 4 |
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( |