Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
  • Loading branch information
wx-csy committed Dec 20, 2024
1 parent 86db0f6 commit c882613
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion mooncake-transfer-engine/src/topology.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
#include <vector>

#ifdef USE_CUDA
#include "cuda_runtime.h"
#include <cuda_runtime.h>
#endif

#include <ctype.h>
Expand Down Expand Up @@ -172,6 +172,8 @@ static std::vector<TopologyEntry> discover_cuda_topology(
std::vector<std::string> preferred_hca;
std::vector<std::string> avail_hca;
for (const auto &hca : all_hca) {
// FIXME: currently we only identify the NICs connected to the same
// PCIe switch/RC with GPU as preferred.
if (get_pci_distance(hca.pci_bus_id.c_str(), pci_bus_id) == 0) {
preferred_hca.push_back(hca.name);
} else {
Expand Down

0 comments on commit c882613

Please sign in to comment.