experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove TF_CPP_VMODULE from NCCL tests since it has no effect when set after

Browse Source 
program start.

PiperOrigin-RevId: 247040886
This commit is contained in:
Ayush Dubey 2019-05-07 10:07:58 -07:00 committed by TensorFlower Gardener
parent c99a4a8610
commit bf1d138211
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/collective_nccl_reducer_test.cc
View File

@ -95,7 +95,6 @@ class NcclReducerTest : public ::testing::Test {
void Init(int num_ranks) { void Init(int num_ranks) {
setenv("NCCL_DEBUG", "INFO", 1 /* replace */); setenv("NCCL_DEBUG", "INFO", 1 /* replace */);
setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */); setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */);
setenv("TF_CPP_VMODULE", "nccl_manager=2", 1 /* replace */);
InitGPUDevices(); InitGPUDevices();
std::vector<std::unique_ptr<Device>> local_devices; std::vector<std::unique_ptr<Device>> local_devices;
std::vector<string> device_names; std::vector<string> device_names;

1
tensorflow/core/nccl/nccl_manager_test.cc
View File

@ -66,7 +66,6 @@ class NcclManagerTest : public ::testing::Test {
static void SetUpTestCase() { static void SetUpTestCase() {
setenv("NCCL_DEBUG", "INFO", 1 /* replace */); setenv("NCCL_DEBUG", "INFO", 1 /* replace */);
setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */); setenv("NCCL_LAUNCH_MODE", "PARALLEL", 1 /* replace */);
setenv("TF_CPP_VMODULE", "nccl_manager=2", 1 /* replace */);
devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices()); devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices());
LOG(INFO) << "Running test with " << devices_->size() << " gpus"; LOG(INFO) << "Running test with " << devices_->size() << " gpus";
} }