From 7c4dd6733590fb59474f6d9e07b1af8c5905fd17 Mon Sep 17 00:00:00 2001 From: jiangyimin Date: Fri, 28 Jun 2019 15:41:54 +0800 Subject: [PATCH] common: change default pcie size to 8 --- byteps/common/nccl_manager.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/byteps/common/nccl_manager.cc b/byteps/common/nccl_manager.cc index 1b9350edb..53f8ff4db 100644 --- 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) {