Nvtx support (#1076)

* NVTX support
2024-02-08 14:08:24 -07:00
parent 5257c753c5
commit 8a442faa12
4 changed files with 20 additions and 4 deletions
@@ -60,6 +60,10 @@ static __inline__ int ncclTypeSize(ncclDataType_t type) {
 #include "alloc.h"
 #include "utils.h"
 #include "param.h"
+#ifdef NVTX_NO_IMPL
 #include "nvtx_stub.h"
+#else
+#include "nvtx.h"
+#endif

 #endif // end include guard