diff options
author | Ke Wen <kwen@nvidia.com> | 2019-07-16 21:28:19 +0300 |
---|---|---|
committer | Ke Wen <kwen@nvidia.com> | 2019-07-17 16:32:33 +0300 |
commit | 4d579e51ccb4d332c0b13783cc979dd7f5cf05d3 (patch) | |
tree | 136f317bef661a811b7435d98cf34c520d97137f | |
parent | 920ae57c147de7a53a9d47d2171be7c10d5e845c (diff) |
Fix NIC distances for 11+ NICs
-rw-r--r-- | src/transport/net.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/transport/net.cc b/src/transport/net.cc index 823caf1..d9559eb 100644 --- a/src/transport/net.cc +++ b/src/transport/net.cc @@ -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) { ncclTvalue_t tvalue = 0; 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 tvalue |= ((score & NET_BITS_PER_IF_MASK)<<(NET_BITS_PER_IF*d)); } |