diff options
Diffstat (limited to 'tests/validation/main.cpp')
-rw-r--r-- | tests/validation/main.cpp | 14 |
1 files changed, 3 insertions, 11 deletions
diff --git a/tests/validation/main.cpp b/tests/validation/main.cpp index 6ec6885159..a6c02ed4dd 100644 --- a/tests/validation/main.cpp +++ b/tests/validation/main.cpp @@ -23,8 +23,8 @@ */ #define BOOST_TEST_ALTERNATIVE_INIT_API +#include "AssetsLibrary.h" #include "Globals.h" -#include "TensorLibrary.h" #include "Utils.h" #include "ValidationProgramOptions.h" #include "ValidationUserConfiguration.h" @@ -46,7 +46,7 @@ namespace arm_compute namespace test { ValidationUserConfiguration user_config; -std::unique_ptr<TensorLibrary> library; +std::unique_ptr<AssetsLibrary> library; } // namespace test } // namespace arm_compute @@ -54,15 +54,7 @@ struct GlobalFixture { GlobalFixture() { - if(user_config.seed.is_set()) - { - library = arm_compute::support::cpp14::make_unique<TensorLibrary>(user_config.path.get(), user_config.seed); - } - else - { - library = arm_compute::support::cpp14::make_unique<TensorLibrary>(user_config.path.get()); - } - + library = arm_compute::support::cpp14::make_unique<AssetsLibrary>(user_config.path.get(), user_config.seed); std::cout << "Seed: " << library->seed() << "\n"; } }; |