Enhance nvtx support. (#5636)

2020-05-06 22:54:24 +08:00
parent 67d267f9da
commit eaf2a00b5c
13 changed files with 98 additions and 87 deletions
--- a/tests/cpp/CMakeLists.txt
+++ b/tests/cpp/CMakeLists.txt
@@ -40,9 +40,9 @@ if (USE_CUDA)
  endif (USE_NCCL)

  if (USE_NVTX)
-    target_include_directories(testxgboost PRIVATE "${NVTX_HEADER_DIR}")
-    target_compile_definitions(testxgboost PRIVATE -DXGBOOST_USE_NVTX=1)
+    enable_nvtx(testxgboost)
  endif (USE_NVTX)
+
  if (MSVC)
    target_compile_options(testxgboost PRIVATE
      $<$<COMPILE_LANGUAGE:CUDA>:-Xcompiler=/utf-8>