diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/CMakeLists.txt | 4 | ||||
-rw-r--r-- | tests/InferenceTestImage.cpp | 6 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index a0e4bf85b5..028fc8283f 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -1,8 +1,6 @@ # UnitTests include(CheckIncludeFiles) -include_directories(SYSTEM "${THIRD_PARTY_INCLUDE_DIRS}") - # Setup the inference test framework set(inference_test_sources ClassifierTestCaseData.hpp @@ -12,9 +10,11 @@ set(inference_test_sources InferenceTest.cpp InferenceTestImage.hpp InferenceTestImage.cpp) + add_library_ex(inferenceTest STATIC ${inference_test_sources}) target_include_directories(inferenceTest PRIVATE ../src/armnnUtils) target_include_directories(inferenceTest PRIVATE ../src/backends) +target_include_directories(inferenceTest PRIVATE ../third-party/stb) if(BUILD_CAFFE_PARSER) macro(CaffeParserTest testName sources) diff --git a/tests/InferenceTestImage.cpp b/tests/InferenceTestImage.cpp index c305d43277..b011e6ac8f 100644 --- a/tests/InferenceTestImage.cpp +++ b/tests/InferenceTestImage.cpp @@ -12,13 +12,13 @@ #include <array> #define STB_IMAGE_IMPLEMENTATION -#include <stb_image.h> +#include <stb/stb_image.h> #define STB_IMAGE_RESIZE_IMPLEMENTATION -#include <stb_image_resize.h> +#include <stb/stb_image_resize.h> #define STB_IMAGE_WRITE_IMPLEMENTATION -#include <stb_image_write.h> +#include <stb/stb_image_write.h> namespace { |