Add 'projects/rccl/' from commit '1f2f9f33bac3e8ecfd84c69af6063d7352c362fc'
git-subtree-dir: projects/rccl git-subtree-mainline:3fd8a0d393git-subtree-split:1f2f9f33ba
This commit is contained in:
@@ -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;
|
||||
Reference in New Issue
Block a user