diff --git a/tensorflow/core/kernels/collective_nccl_reducer_test.cc b/tensorflow/core/kernels/collective_nccl_reducer_test.cc index 7348019e056..00dfa722b57 100644 --- a/tensorflow/core/kernels/collective_nccl_reducer_test.cc +++ b/tensorflow/core/kernels/collective_nccl_reducer_test.cc @@ -95,7 +95,6 @@ class NcclReducerTest : public ::testing::Test { void Init(int num_ranks) { setenv("NCCL_DEBUG", "INFO", 1 /* replace */); setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */); - setenv("TF_CPP_VMODULE", "nccl_manager=2", 1 /* replace */); InitGPUDevices(); std::vector> local_devices; std::vector device_names; diff --git a/tensorflow/core/nccl/nccl_manager_test.cc b/tensorflow/core/nccl/nccl_manager_test.cc index fcca62bdea0..06564ee8020 100644 --- a/tensorflow/core/nccl/nccl_manager_test.cc +++ b/tensorflow/core/nccl/nccl_manager_test.cc @@ -66,7 +66,6 @@ class NcclManagerTest : public ::testing::Test { static void SetUpTestCase() { setenv("NCCL_DEBUG", "INFO", 1 /* replace */); setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */); - setenv("TF_CPP_VMODULE", "nccl_manager=2", 1 /* replace */); devices_ = new std::vector>(GetGPUDevices()); LOG(INFO) << "Running test with " << devices_->size() << " gpus"; }