Fix Windows builds

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

Author: cyyever

libkineto/CMakeLists.txt

@@ -71,20 +71,26 @@ if(NOT LIBKINETO_NOCUPTI)
       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}")