Enable MSCCLPP use in CPX mode (#1355)
This PR enables the use of MSCCLPP in CPX mode for 8 GPUs.
[ROCm/rccl commit: d13f9c44f5]
Cette révision appartient à :
@@ -63,7 +63,10 @@ if(ENABLE_MSCCLPP)
|
||||
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
|
||||
)
|
||||
endif()
|
||||
|
||||
execute_process(
|
||||
COMMAND git apply ${CMAKE_CURRENT_SOURCE_DIR}/ext-src/cpx.patch
|
||||
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
|
||||
)
|
||||
message(STATUS "Building mscclpp only for gfx942.")
|
||||
|
||||
mscclpp_cmake_arg(CMAKE_PREFIX_PATH)
|
||||
|
||||
@@ -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;
|
||||
Référencer dans un nouveau ticket
Bloquer un utilisateur