diff options
Diffstat (limited to 'tests/benchmark_new/NEON/GEMM.cpp')
| -rw-r--r-- | tests/benchmark_new/NEON/GEMM.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/benchmark_new/NEON/GEMM.cpp b/tests/benchmark_new/NEON/GEMM.cpp index 640ec1c032..58d5864ba6 100644 --- a/tests/benchmark_new/NEON/GEMM.cpp +++ b/tests/benchmark_new/NEON/GEMM.cpp @@ -30,6 +30,7 @@ #include "framework/datasets/Datasets.h" #include "tests/TypePrinter.h" #include "tests/datasets_new/GoogLeNetGEMMDataset.h" +#include "tests/datasets_new/MatrixMultiplyGEMMDataset.h" #include "tests/fixtures_new/GEMMFixture.h" namespace arm_compute @@ -38,7 +39,7 @@ namespace test { namespace { -auto data_types = framework::dataset::make("DataType", +const auto data_types = framework::dataset::make("DataType", { #if ARM_COMPUTE_ENABLE_FP16 DataType::F16, @@ -52,7 +53,8 @@ using NEGEMMFixture = GEMMFixture<Tensor, NEGEMM>; TEST_SUITE(NEON) -REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetGEMM, NEGEMMFixture, framework::DatasetMode::ALL, framework::dataset::combine(datasets::GoogLeNetGEMMDataset(), std::move(data_types))); +REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetGEMM, NEGEMMFixture, framework::DatasetMode::ALL, framework::dataset::combine(datasets::GoogLeNetGEMMDataset(), data_types)); +REGISTER_FIXTURE_DATA_TEST_CASE(MatrixMultiplyGEMM, NEGEMMFixture, framework::DatasetMode::ALL, framework::dataset::combine(datasets::MatrixMultiplyGEMMDataset(), data_types)); TEST_SUITE_END() } // namespace test |