diff options
| -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( |