Nvtx support (#1076)

* NVTX support
Esse commit está contido em:
Bertan Dogancay
2024-02-08 14:08:24 -07:00
commit de GitHub
commit 8a442faa12
4 arquivos alterados com 20 adições e 4 exclusões
+4
Ver Arquivo
@@ -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