Fix Windows builds

Signed-off-by: cyy <[email protected]>

Author: cyyever

libkineto/CMakeLists.txt

@@ -65,26 +65,32 @@ if (CUDAToolkit_LIBRARY_ROOT)
       "${CUDAToolkit_LIBRARY_ROOT}/lib64"
   )
 endif()
-if(NOT LIBKINETO_NOCUPTI)
+if(NOT LIBKINETO_NOCUPTI AND CUDAToolkit_LIBRARY_ROOT)
   if(NOT CUPTI_INCLUDE_DIR)
     find_path(CUPTI_INCLUDE_DIR cupti.h PATHS ${CUDA_PATHS}
       NO_DEFAULT_PATH)
   endif()
 
-  if(USE_CUPTI_SO)
-    if(NOT CUDA_cupti_LIBRARY)
-      find_library(CUDA_cupti_LIBRARY cupti PATHS ${CUDA_PATHS}
-        NO_DEFAULT_PATH)
-    endif()
-    set(cupti_LIBRARY ${CUDA_cupti_LIBRARY})
-  else()
-    if(NOT CUDA_cupti_static_LIBRARY)
-      find_library(CUDA_cupti_static_LIBRARY cupti_static PATHS ${CUDA_PATHS}
-        NO_DEFAULT_PATH)
+  if(NOT USE_CUPTI_SO AND NOT CUDA_cupti_static_LIBRARY)
+    set(cupti_static_libs "cupti_static")
+    if(WIN32)
+      list(APPEND cupti_static_libs "cupti")
+      find_library(CUDA_cupti_static_LIBRARY cupti_static cupti PATHS ${CUDA_PATHS}
+          NO_DEFAULT_PATH)
     endif()
+  endif()
+  if(CUDA_cupti_static_LIBRARY)
     set(cupti_LIBRARY ${CUDA_cupti_static_LIBRARY})
   endif()
 
+  if(USE_CUPTI_SO AND NOT CUDA_cupti_LIBRARY)
+    find_library(CUDA_cupti_LIBRARY cupti PATHS ${CUDA_PATHS}
+        NO_DEFAULT_PATH)
+  endif()
+  if(CUDA_cupti_LIBRARY)
+    set(cupti_LIBRARY ${CUDA_cupti_LIBRARY})
+  endif()
+
   if(cupti_LIBRARY AND CUPTI_INCLUDE_DIR)
     message(STATUS "  CUPTI_INCLUDE_DIR = ${CUPTI_INCLUDE_DIR}")
     message(STATUS "  CUDA_cupti_LIBRARY = ${CUDA_cupti_LIBRARY}")
@@ -129,7 +135,7 @@ if(NOT LIBKINETO_NOCUPTI)
     message(STATUS "Could not find CUPTI library")
   endif()
 endif()
-if(NOT TARGET CUDA::nvperf_host)
+if(NOT TARGET CUDA::nvperf_host AND CUDAToolkit_LIBRARY_ROOT)
   find_library(CUDA_NVPERF_HOST_LIB_PATH nvperf_host PATHS ${CUDA_PATHS}
         NO_DEFAULT_PATH)
   if(CUDA_NVPERF_HOST_LIB_PATH)