update build rccl-tests infrastructure and add getAlgoProtoChannels support (#2212)

projects/rocprofiler-systems/examples/rccl/rccl-tests/src/all_gather.cpp

@@ -7,6 +7,7 @@
 
 #include "common.h"
 #include "cuda_runtime.h"
+#include "rccl_compat.h"
 
 void
 AllGatherGetCollByteCount(size_t* sendcount, size_t* recvcount, size_t* paramcount,
@@ -47,6 +48,16 @@ AllGatherInitData(struct threadArgs* args, ncclDataType_t type, ncclRedOp_t op,
     return testSuccess;
 }
 
+testResult_t
+AllGatherGetAlgoProtoChannels(ncclComm_t comm, size_t count, ncclDataType_t type,
+                              int* algo, int* proto, int* nchannels)
+{
+    if(rcclTestsGetAlgoInfo == NULL) return testInternalError;
+    NCCLCHECK(rcclTestsGetAlgoInfo(comm, ncclFuncAllGather, count, type, 0, 0, 1, algo,
+                                   proto, nchannels));
+    return testSuccess;
+}
+
 void
 AllGatherGetBw(size_t count, int typesize, double sec, double* algBw, double* busBw,
                int nranks)
@@ -60,14 +71,17 @@ AllGatherGetBw(size_t count, int typesize, double sec, double* algBw, double* bu
 
 testResult_t
 AllGatherRunColl(void* sendbuff, void* recvbuff, size_t count, ncclDataType_t type,
-                 ncclRedOp_t op, int root, ncclComm_t comm, cudaStream_t stream)
+                 ncclRedOp_t op, int root, ncclComm_t comm, cudaStream_t stream,
+                 void* bias = nullptr)
 {
+    (void) bias;
     NCCLCHECK(ncclAllGather(sendbuff, recvbuff, count, type, comm, stream));
     return testSuccess;
 }
 
-struct testColl allGatherTest = { "AllGather", AllGatherGetCollByteCount,
-                                  AllGatherInitData, AllGatherGetBw, AllGatherRunColl };
+struct testColl allGatherTest = { "AllGather",       AllGatherGetCollByteCount,
+                                  AllGatherInitData, AllGatherGetBw,
+                                  AllGatherRunColl,  AllGatherGetAlgoProtoChannels };
 
 void
 AllGatherGetBuffSize(size_t* sendcount, size_t* recvcount, size_t count, int nranks)