saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Restore cuda in test

Browse source
This commit is contained in:
Ryan Hill 2021-04-21 20:28:03 -07:00
parent e74de85a53
commit 9f1c2ed767
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cmake/onnxruntime_unittests.cmake
View file

@ -929,10 +929,9 @@ if (onnxruntime_BUILD_SHARED_LIB)
if(NOT WIN32)
list(APPEND onnxruntime_shared_lib_test_LIBS nsync_cpp)
endif()
# TODO: Cuda is a separate shared library, can't just add .cu stuff to the main library
# if (onnxruntime_USE_CUDA)
# list(APPEND onnxruntime_shared_lib_test_LIBS onnxruntime_test_cuda_ops_lib cudart)
# endif()
if (onnxruntime_USE_CUDA)
list(APPEND onnxruntime_shared_lib_test_LIBS onnxruntime_test_cuda_ops_lib cudart)
endif()
if (CMAKE_SYSTEM_NAME STREQUAL "Android")
list(APPEND onnxruntime_shared_lib_test_LIBS ${android_shared_libs})
endif()