Fix collective trace when rccl is configured (#1056)
* Fix collective trace when rccl is configured
[ROCm/rccl commit: c4dbf8a914]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
8b8179a689
Коммит
56482a8be8
@@ -21,6 +21,9 @@
|
||||
#define NCCL_NUM_FUNCTIONS 5 // SendRecv and AllToAllPivot not included for now
|
||||
typedef enum { ncclFuncBroadcast, ncclFuncReduce, ncclFuncAllGather, ncclFuncReduceScatter, ncclFuncAllReduce, ncclFuncSendRecv, ncclFuncSend, ncclFuncRecv, ncclFuncAllToAllPivot, ncclNumFuncs} ncclFunc_t;
|
||||
extern const char* ncclFuncStr[NCCL_NUM_FUNCTIONS+2];
|
||||
#ifdef ENABLE_COLLTRACE
|
||||
extern const char* funcNames[FUNC_INDEX_TOTAL];
|
||||
#endif
|
||||
|
||||
#define NCCL_NUM_ALGORITHMS 6 // Tree/Ring/CollNet*
|
||||
#define NCCL_ALGO_TREE 0
|
||||
|
||||
Ссылка в новой задаче
Block a user