Fix collective trace when rccl is configured (#1056)

* Fix collective trace when rccl is configured

[ROCm/rccl commit: c4dbf8a914]
Этот коммит содержится в:
Bertan Dogancay
2024-01-22 09:26:44 -07:00
коммит произвёл GitHub
родитель 8b8179a689
Коммит 56482a8be8
5 изменённых файлов: 28 добавлений и 68 удалений
+3
Просмотреть файл
@@ -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