vstrt/{trtexec}/CMakeLists.txt: linking cublas

AmusementClub · Oct 22, 2023 · 0fe13e7 · 0fe13e7
1 parent a4ae49e
commit 0fe13e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/vstrt/CMakeLists.txt b/vstrt/CMakeLists.txt
@@ -46,13 +46,13 @@ if (USE_NVINFER_PLUGIN)
         target_link_libraries(vstrt PRIVATE ${CUDNN_LIB} CUDA::cublas)
 
         if (CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
-            target_link_options(vstrt PRIVATE "/DELAYLOAD:cublas64_11.dll" "/DELAYLOAD:cudnn64_8.dll")
+            target_link_options(vstrt PRIVATE "/DELAYLOAD:cublas64_12.dll" "/DELAYLOAD:cudnn64_8.dll")
         endif()
     else()
-        target_link_libraries(vstrt PRIVATE nvinfer_plugin)
+        target_link_libraries(vstrt PRIVATE nvinfer_plugin CUDA::cublas)
 
         if (CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
-            target_link_options(vstrt PRIVATE "/DELAYLOAD:nvinfer_plugin.dll")
+            target_link_options(vstrt PRIVATE "/DELAYLOAD:nvinfer_plugin.dll" "/DELAYLOAD:cublas64_12.dll")
         endif()
     endif()
 endif()

diff --git a/vstrt/trtexec/CMakeLists.txt b/vstrt/trtexec/CMakeLists.txt
@@ -34,6 +34,7 @@ target_link_libraries(trtexec PRIVATE
     nvinfer_plugin
     nvparsers
     nvonnxparser
+    CUDA::cublas
 )
 
 install(TARGETS trtexec RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})