From de49a77074e884758e689ff0204f066eac7aae46 Mon Sep 17 00:00:00 2001
From: Jack Snyder <jsnyder@prom.nvidia.com>
Date: Wed, 5 Aug 2020 13:34:32 -0700
Subject: [PATCH] Setting type when gpu sub node is discovered

---
 src/graph/topo.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/graph/topo.cc b/src/graph/topo.cc
index a917340..ed4bd23 100644
--- a/src/graph/topo.cc
+++ b/src/graph/topo.cc
@@ -320,6 +320,7 @@ ncclResult_t ncclTopoAddPci(struct ncclXmlNode* xmlPci, struct ncclTopoSystem* s
   struct ncclXmlNode* xmlGpu = NULL;
   NCCLCHECK(xmlGetSub(xmlPci, "gpu", &xmlGpu));
   if (xmlGpu != NULL) {
+    type = GPU;
     int index;
     NCCLCHECK(xmlGetAttrIndex(xmlGpu, "rank", &index));
     if (index == -1) return ncclSuccess;