Skip to content

Commit b502b3b

Browse files
authored
Modernize nvtx3 Include (#4774)
## Summary https://github.com/NVIDIA/NVTX?tab=readme-ov-file#example-visualize-loop-iterations-on-a-timeline ## Additional background ## Checklist The proposed changes: - [ ] fix a bug or incorrect behavior in AMReX - [ ] add new capabilities to AMReX - [ ] changes answers in the test suite to more than roundoff level - [ ] are likely to significantly affect the results of downstream AMReX users - [ ] include documentation in the code and/or rst files, if appropriate
1 parent e727197 commit b502b3b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

Src/Base/AMReX_GpuDevice.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,9 @@
2020
#if defined(AMREX_USE_CUDA)
2121
#include <cuda_profiler_api.h>
2222
#if defined(AMREX_PROFILING) || defined (AMREX_TINY_PROFILING)
23-
#if __has_include(<nvtx3/nvToolsExt.h>)
23+
#if __has_include(<nvtx3/nvtx3.hpp>)
24+
# include <nvtx3/nvtx3.hpp>
25+
#elif __has_include(<nvtx3/nvToolsExt.h>)
2426
# include <nvtx3/nvToolsExt.h>
2527
#else
2628
# include <nvToolsExt.h>

Src/Base/AMReX_TinyProfiler.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@
1717
#endif
1818

1919
#ifdef AMREX_USE_CUDA
20-
#if __has_include(<nvtx3/nvToolsExt.h>)
20+
#if __has_include(<nvtx3/nvtx3.hpp>)
21+
# include <nvtx3/nvtx3.hpp>
22+
#elif __has_include(<nvtx3/nvToolsExt.h>)
2123
# include <nvtx3/nvToolsExt.h>
2224
#else
2325
# include <nvToolsExt.h>

0 commit comments

Comments
 (0)