Move cupti exception check from PyTorch to kineto

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

Author: cyyever

libkineto/CMakeLists.txt

@@ -4,7 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-cmake_minimum_required(VERSION 3.22 FATAL_ERROR)
+cmake_minimum_required(VERSION 3.24 FATAL_ERROR)
 project(kineto VERSION 0.1 LANGUAGES CXX C)
 
 #install libraries into correct locations on all platforms
@@ -79,8 +79,32 @@ if(NOT LIBKINETO_NOCUPTI)
     if(NOT TARGET CUDA::cupti)
       add_library(CUDA::cupti INTERFACE IMPORTED)
       target_link_libraries(CUDA::cupti INTERFACE "${CUDA_cupti_LIBRARY}")
+    else()
+      set(CUDA_cupti_LIBRARY "$<TARGET_FILE:CUDA::cupti>")
     endif()
     target_include_directories(CUDA::cupti INTERFACE "${CUPTI_INCLUDE_DIR}")
+
+    if("${CUDA_cupti_LIBRARY}" MATCHES "[.]a")
+      include(CheckCXXSourceRuns)
+      set(CMAKE_REQUIRED_LIBRARIES ${CMAKE_REQUIRED_LIBRARIES} "dl" "pthread" $<LINK_LIBRARY:WHOLE_ARCHIVE,"${CUDA_cupti_LIBRARY}">)
+
+      check_cxx_source_runs("#include <stdexcept>
+  int main() {
+  try {
+    throw std::runtime_error(\"error\");
+  } catch (...) {
+    return 0;
+  }
+  return 1;
+  }" EXCEPTIONS_WORK)
+      set(CMAKE_REQUIRED_LINK_OPTIONS "")
+      if(NOT EXCEPTIONS_WORK)
+        message(FATAL_ERROR
+          "Detected that statically linking against CUPTI causes exceptions to stop working. "
+          "See https://github.com/pytorch/pytorch/issues/57744 for more details. ")
+      endif()
+    endif()
+
   else()
     set(LIBKINETO_NOCUPTI ON CACHE BOOL "" FORCE)
     message(STATUS "Could not find CUPTI library")