yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix NIC distances for 11+ NICs

Browse Source
This commit is contained in:
Ke Wen 2019-07-16 11:28:19 -07:00
parent 920ae57c14
commit 4d579e51cc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transport/net.cc
View File

@ -28,7 +28,7 @@ static_assert(sizeof(ncclTvalue_t)*8 >= NET_MAX_IFS*NET_BITS_PER_IF, "NET_MAX_IF
static ncclTvalue_t getTvalue(short* distances, int ndev) { static ncclTvalue_t getTvalue(short* distances, int ndev) {
ncclTvalue_t tvalue = 0; ncclTvalue_t tvalue = 0;
for (int d=0; d<ndev; d++) { for (int d=0; d<ndev; d++) {
int score = 1 + PATH_SYS - distances[d]; ncclTvalue_t score = 1 + PATH_SYS - distances[d];
// Keep 3 bits of score info per dev // Keep 3 bits of score info per dev
tvalue |= ((score & NET_BITS_PER_IF_MASK)<<(NET_BITS_PER_IF*d)); tvalue |= ((score & NET_BITS_PER_IF_MASK)<<(NET_BITS_PER_IF*d));
} }