adding all nccl apis to api_support to enable rccl tracing by rocprofv3 (#1297)

* adding all nccl apis to api_support to enable rccl tracing by rocprofv3

Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
Co-authored-by: Jonathan R. Madsen <jonathanrmadsen@gmail.com>
This commit is contained in:
mberenjk
2024-08-22 12:36:07 -05:00
committato da GitHub
parent 93b0c7418f
commit db840f024e
11 ha cambiato i file con 945 aggiunte e 38 eliminazioni
+3 -2
Vedi File
@@ -11,6 +11,7 @@
#include "enqueue.h"
#include "transport.h"
#include "channel.h"
#include "api_trace.h"
#include <assert.h>
#include "msccl/msccl_lifecycle.h"
@@ -83,7 +84,7 @@ ncclResult_t ncclAsyncJobComplete(struct ncclAsyncJob* job) {
}
NCCL_API(ncclResult_t, ncclGroupStart);
ncclResult_t ncclGroupStart() {
ncclResult_t ncclGroupStart_impl() {
ncclResult_t ret = ncclSuccess;
NVTX3_FUNC_RANGE_IN(nccl_domain);
@@ -101,7 +102,7 @@ ncclResult_t ncclGroupStartInternal() {
}
NCCL_API(ncclResult_t, ncclGroupEnd);
ncclResult_t ncclGroupEnd() {
ncclResult_t ncclGroupEnd_impl() {
ncclResult_t ret = ncclSuccess;
NVTX3_FUNC_RANGE_IN(nccl_domain);
NCCLCHECKGOTO(ncclGroupEndInternal(), ret, exit);