Skip to content

Commit d41df94

Browse files
committed
feat(topology): add GPU entries alongside CUDA entries in topology discovery
Signed-off-by: staryxchen <[email protected]>
1 parent 392fea7 commit d41df94

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

mooncake-transfer-engine/src/topology.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,10 @@ static std::vector<TopologyEntry> discoverCudaTopology(
207207
TopologyEntry{.name = "cuda:" + std::to_string(i),
208208
.preferred_hca = std::move(preferred_hca),
209209
.avail_hca = std::move(avail_hca)});
210+
topology.push_back(
211+
TopologyEntry{.name = "gpu:" + std::to_string(i),
212+
.preferred_hca = std::move(preferred_hca),
213+
.avail_hca = std::move(avail_hca)});
210214
}
211215
return topology;
212216
}

0 commit comments

Comments
 (0)