AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add 8P6L multi-node models (#239)

Browse Source 
[ROCm/rccl commit: d5f90e19b5]
This commit is contained in:
Wenkai Du
2020-07-21 14:10:36 -07:00
committed by GitHub
parent 56d8c7c893
commit f604fc774e
15 changed files with 830 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/include/graph.h
+2 -2
View File
@@ -42,6 +42,8 @@ ncclResult_t ncclTopoSetAffinity(struct ncclTopoSystem* system, int rank);
#define NCCL_TOPO_CPU_VENDOR_AMD 2
#define NCCL_TOPO_CPU_TYPE_BDW 1
#define NCCL_TOPO_CPU_TYPE_SKL 2
#define NCCL_TOPO_CPU_TYPE_ZEN 3
#define NCCL_TOPO_CPU_TYPE_ROME 4
ncclResult_t ncclTopoCpuType(struct ncclTopoSystem* system, int* arch, int* vendor, int* model);
#define NCCL_TOPO_MAX_NODES 256
@@ -101,6 +103,4 @@ ncclResult_t ncclTopoTuneModel(struct ncclComm* comm, int minCompCap, int maxCom
#include "info.h"
ncclResult_t ncclTopoGetAlgoTime(struct ncclInfo* info, int algorithm, int protocol, float* time);
ncclResult_t parseChordalRing(struct ncclTopoSystem* system, char **str);
#endif