Merge remote-tracking branch 'nccl/master' into HEAD

[ROCm/rccl commit: e1cb45ff22]
2023-02-04 01:43:38 +00:00
parent f7982e9bed 2840ac0139
commit c76bc214c8
62 changed files with 3799 additions and 1447 deletions
@@ -7,9 +7,14 @@
 #ifndef NCCL_NVTX_STUB_H_
 #define NCCL_NVTX_STUB_H_

+#include <nvtx3/nvToolsExtPayload.h>
+
 struct nccl_domain{static constexpr char const* name{"NCCL"};};

 #define NVTX3_FUNC_RANGE_IN(domain)
 #define nvtxNameOsThreadA(syscall, thread)
+#define NVTX3_FUNC_WITH_PARAMS(ID, S, P)
+
+#define NVTX_PAYLOAD_ENTRY_NCCL_REDOP 11

 #endif