diff options
Diffstat (limited to 'delegate/CMakeLists.txt')
-rw-r--r-- | delegate/CMakeLists.txt | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/delegate/CMakeLists.txt b/delegate/CMakeLists.txt index f6a5e512fa..d35fd4e7b6 100644 --- a/delegate/CMakeLists.txt +++ b/delegate/CMakeLists.txt @@ -233,7 +233,7 @@ if(BUILD_UNIT_TESTS) else() endif() - if (BUILD_CLASSIC_DELEGATE) # For Opaque Delegate Unit Tests add OR BUILD_ARMNN_TFLITE_OPAQUE_DELEGATE here + if (BUILD_CLASSIC_DELEGATE OR BUILD_OPAQUE_DELEGATE) add_executable(DelegateUnitTests ${armnnDelegate_unittest_sources}) target_include_directories(DelegateUnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}") @@ -241,7 +241,14 @@ if(BUILD_UNIT_TESTS) # Add half library from armnn third-party libraries target_link_libraries(DelegateUnitTests PRIVATE thirdparty_headers) - target_link_libraries(DelegateUnitTests PRIVATE armnnDelegate) + if (BUILD_CLASSIC_DELEGATE) + target_link_libraries(DelegateUnitTests PRIVATE armnnDelegate) + target_include_directories(DelegateUnitTests PUBLIC ${PROJECT_SOURCE_DIR}) + endif() + if (BUILD_OPAQUE_DELEGATE) + target_link_libraries(DelegateUnitTests PRIVATE armnnOpaqueDelegate) + target_include_directories(DelegateUnitTests PUBLIC ${PROJECT_SOURCE_DIR}) + endif() target_link_libraries(DelegateUnitTests PRIVATE Armnn::armnnUtils) target_link_libraries(DelegateUnitTests PRIVATE profiling_library_headers) |