Add 'projects/rccl/' from commit '1f2f9f33bac3e8ecfd84c69af6063d7352c362fc'

git-subtree-dir: projects/rccl
git-subtree-mainline: 3fd8a0d393
git-subtree-split: 1f2f9f33ba
This commit is contained in:
Ameya Keshava Mallya
2025-12-11 20:46:05 +00:00
682 zmienionych plików z 293338 dodań i 0 usunięć
+12
Wyświetl plik
@@ -0,0 +1,12 @@
diff --git a/src/numa.cc b/src/numa.cc
index d72c99e..16c903d 100644
--- a/src/numa.cc
+++ b/src/numa.cc
@@ -26,6 +26,7 @@ namespace mscclpp {
MSCCLPP_API_CPP int getDeviceNumaNode(int cudaDev) {
std::string busId = getBusId(cudaDev);
+ busId[busId.length() - 1] = '0';
std::string file_str = "/sys/bus/pci/devices/" + busId + "/numa_node";
std::ifstream file(file_str);
int numaNode;