Skip to content

Commit 6688f51

Browse files
committed
Fix: switch to nvtx3 headers when CUDA_VERSION >= 12090
Signed-off-by:Tianxiang Wang<[email protected]>, Contributed under MetaX Integrated Circuits (Shanghai) Co., Ltd.
1 parent 8f7d319 commit 6688f51

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

source/source_base/timer.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,11 @@
1515
#include "source_base/formatter.h"
1616

1717
#if defined(__CUDA) && defined(__USE_NVTX)
18-
#include <nvToolsExt.h>
18+
#if CUDA_VERSION < 12090
19+
#include "nvToolsExt.h"
20+
#else
21+
#include "nvtx3/nvToolsExt.h"
22+
#endif
1923
#include "source_io/module_parameter/parameter.h"
2024
#endif
2125

0 commit comments

Comments
 (0)