From 9f1c2ed76733bfd4ed704630723c6a305d7b4464 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Wed, 21 Apr 2021 20:28:03 -0700 Subject: [PATCH] Restore cuda in test --- cmake/onnxruntime_unittests.cmake | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/cmake/onnxruntime_unittests.cmake b/cmake/onnxruntime_unittests.cmake index dffc611268..fa4fb63d98 100644 --- a/cmake/onnxruntime_unittests.cmake +++ b/cmake/onnxruntime_unittests.cmake @@ -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()