From bf1d138211f2ca1a23923ba44b2a234417e46adb Mon Sep 17 00:00:00 2001 From: Ayush Dubey Date: Tue, 7 May 2019 10:07:58 -0700 Subject: [PATCH] Remove TF_CPP_VMODULE from NCCL tests since it has no effect when set after program start. PiperOrigin-RevId: 247040886 --- tensorflow/core/kernels/collective_nccl_reducer_test.cc | 1 - tensorflow/core/nccl/nccl_manager_test.cc | 1 - 2 files changed, 2 deletions(-) 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"; }