microsoft
/
msccl
зеркало из https://github.com/microsoft/msccl.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

done on 64 gpus

This commit is contained in:
Saeed Maleki 2021-03-31 01:17:19 +00:00
Родитель f7837e5614
Коммит 7af77c04b8
2 изменённых файлов: 4 добавлений и 5 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

5
src/graph/topo.cc
Просмотреть файл

@ -611,10 +611,9 @@ ncclResult_t scklGetAlgoFromXMLAndSetComm(struct ncclComm* comm) {
char* str = getenv("SCKL_XML_FILE"); char* str = getenv("SCKL_XML_FILE");
if (str){ if (str){
INFO(NCCL_ENV, "SCKL_XML_FILE set by environment to %s", str); INFO(NCCL_ENV, "SCKL_XML_FILE set by environment to %s", str);
struct ncclXml xml_alloc; struct ncclXml* xml;
struct ncclXml* xml = &xml_alloc;
// NCCLCHECK(ncclCalloc(&xml, 1)); NCCLCHECK(ncclCalloc(&xml, 1));
NCCLCHECK(scklGetXmlAlgoFromFile(str, xml)); NCCLCHECK(scklGetXmlAlgoFromFile(str, xml));
int rank = comm->rank; int rank = comm->rank;

4
src/graph/xml.h
Просмотреть файл

@ -11,8 +11,8 @@
// A few constraints to make the implementation easy // A few constraints to make the implementation easy
#define MAX_STR_LEN 255 #define MAX_STR_LEN 255
#define MAX_ATTR_COUNT 16 #define MAX_ATTR_COUNT 16
#define MAX_SUBS 32 #define MAX_SUBS 128
#define MAX_NODES 1024 #define MAX_NODES 16384
#define NODE_TYPE_NONE 0 #define NODE_TYPE_NONE 0
#define NODE_TYPE_OPEN 1 #define NODE_TYPE_OPEN 1