From 4d579e51ccb4d332c0b13783cc979dd7f5cf05d3 Mon Sep 17 00:00:00 2001
From: Ke Wen <kwen@nvidia.com>
Date: Tue, 16 Jul 2019 11:28:19 -0700
Subject: [PATCH] Fix NIC distances for 11+ NICs

---
 src/transport/net.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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));
   }