common: change default pcie size to 8

bytedance · Jun 28, 2019 · 7c4dd67 · 7c4dd67
1 parent 9733f76
commit 7c4dd67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/byteps/common/nccl_manager.cc b/byteps/common/nccl_manager.cc
@@ -131,7 +131,7 @@ void NcclManager::InitGlobalEnv() { // init all global env/param here
     BPS_LOG(DEBUG) << "nccl_group_size" << " set to " << _nccl_group_size;
 
     _nccl_pcie_size = (getenv("BYTEPS_PCIE_SWITCH_SIZE") ?
-                       atoi(getenv("BYTEPS_PCIE_SWITCH_SIZE")) : 4);
+                       atoi(getenv("BYTEPS_PCIE_SWITCH_SIZE")) : 8);
     auto local_size = BytePSGlobal::GetLocalSize();
     _nccl_pcie_num = local_size / _nccl_pcie_size;
     if (!_nccl_pcie_num) {