From a66ab6863038627a205bb136bbc0170299ce85f3 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

[ROCm/rccl commit: 4d579e51ccb4d332c0b13783cc979dd7f5cf05d3]
---
 projects/rccl/src/transport/net.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/rccl/src/transport/net.cc b/projects/rccl/src/transport/net.cc
index 823caf1cc4..d9559eb9bc 100644
--- a/projects/rccl/src/transport/net.cc
+++ b/projects/rccl/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));
   }