pytorch
diff --git a/‎caffe2/CMakeLists.txt
Copy file name to clipboardExpand all lines: caffe2/CMakeLists.txt
+2-14Lines changed: 2 additions & 14 deletions b/‎caffe2/CMakeLists.txt
Copy file name to clipboardExpand all lines: caffe2/CMakeLists.txt
+2-14Lines changed: 2 additions & 14 deletions
@@ -1067,21 +1067,9 @@ elseif(USE_CUDA)
         torch_cuda
     )
     if($ENV{ATEN_STATIC_CUDA})
-      if(CUDAToolkit_VERSION_MAJOR LESS_EQUAL 11)
-        target_link_libraries(torch_cuda_linalg PRIVATE
-            CUDA::cusolver_static
-            ${CUDAToolkit_LIBRARY_DIR}/liblapack_static.a     # needed for libcusolver_static
-        )
-      elseif(CUDAToolkit_VERSION_MAJOR GREATER_EQUAL 12)
-        target_link_libraries(torch_cuda_linalg PRIVATE
-            CUDA::cusolver_static
-            ${CUDAToolkit_LIBRARY_DIR}/libcusolver_lapack_static.a     # needed for libcusolver_static
-        )
-      endif()
+      target_link_libraries(torch_cuda_linalg PRIVATE CUDA::cusolver_static)
     else()
-      target_link_libraries(torch_cuda_linalg PRIVATE
-          CUDA::cusolver
-      )
+      target_link_libraries(torch_cuda_linalg PRIVATE CUDA::cusolver)
     endif()
     set_source_files_properties(${CMAKE_CURRENT_SOURCE_DIR}/../aten/src/ATen/native/cuda/LinearAlgebraStubs.cpp PROPERTIES COMPILE_FLAGS "-DBUILD_LAZY_CUDA_LINALG")
     install(TARGETS torch_cuda_linalg DESTINATION "${TORCH_INSTALL_LIB_DIR}")