Fix Windows builds

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

Author: cyyever

libkineto/CMakeLists.txt

@@ -71,6 +71,19 @@ if(NOT LIBKINETO_NOCUPTI)
       NO_DEFAULT_PATH)
   endif()
 
+  if(NOT USE_CUPTI_SO AND NOT CUDA_cupti_static_LIBRARY)
+      find_library(CUDA_cupti_static_LIBRARY cupti_static PATHS ${CUDA_PATHS}
+        NO_DEFAULT_PATH)
+    endif()
+  endif()
+  if(NOT CUDA_cupti_static_LIBRARY)
+    if(WIN32)
+    message(WARNING "CUPTI is not static on Windows, switch to shared library")
+    set(USE_CUPTI_SO OFF)
+  else()
+    set(cupti_LIBRARY ${CUDA_cupti_static_LIBRARY})
+  endif()
+
   if(USE_CUPTI_SO)
     if(NOT CUDA_cupti_LIBRARY)
       find_library(CUDA_cupti_LIBRARY cupti PATHS ${CUDA_PATHS}