Add Pivot AllToAll algorithm for Rome model (#503)
* add a2a pivot interface * remove debug info * address comments * fix bug * remove custom script * address comments * fix bug
Этот коммит содержится в:
@@ -31,7 +31,7 @@
|
||||
#include "utils.h"
|
||||
#include "rocm_smi/rocm_smi.h"
|
||||
|
||||
const char* ncclFuncStr[NCCL_NUM_FUNCTIONS+1] = { "Broadcast", "Reduce", "AllGather", "ReduceScatter", "AllReduce", "SendRecv" };
|
||||
const char* ncclFuncStr[NCCL_NUM_FUNCTIONS+2] = { "Broadcast", "Reduce", "AllGather", "ReduceScatter", "AllReduce", "SendRecv", "AllToAllPivot" };
|
||||
const char* ncclAlgoStr[NCCL_NUM_ALGORITHMS] = { "Tree", "Ring", "CollNet" };
|
||||
const char* ncclProtoStr[NCCL_NUM_PROTOCOLS] = { "LL", "LL128", "Simple" };
|
||||
|
||||
|
||||
Ссылка в новой задаче
Block a user