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

[ROCm/rccl commit: de49a77074e884758e689ff0204f066eac7aae46]
---
 projects/rccl/src/graph/topo.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/rccl/src/graph/topo.cc b/projects/rccl/src/graph/topo.cc
index a91734020f..ed4bd2322f 100644
--- a/projects/rccl/src/graph/topo.cc
+++ b/projects/rccl/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;