diff options
-rw-r--r-- | tests/benchmark_new/CL/GEMM.cpp | 3 | ||||
-rw-r--r-- | tests/benchmark_new/NEON/GEMM.cpp | 3 | ||||
-rw-r--r-- | tests/fixtures_new/GEMMFixture.h | 6 |
3 files changed, 9 insertions, 3 deletions
diff --git a/tests/benchmark_new/CL/GEMM.cpp b/tests/benchmark_new/CL/GEMM.cpp index a2fd09d059..9b35285e7c 100644 --- a/tests/benchmark_new/CL/GEMM.cpp +++ b/tests/benchmark_new/CL/GEMM.cpp @@ -27,6 +27,7 @@ #include "arm_compute/runtime/CL/functions/CLGEMM.h" #include "framework/Macros.h" #include "framework/datasets/Datasets.h" +#include "tests/CL/CLAccessor.h" #include "tests/TypePrinter.h" #include "tests/datasets_new/MatrixMultiplyGEMMDataset.h" #include "tests/datasets_new/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1GEMMDataset.h" @@ -41,7 +42,7 @@ namespace const auto data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32 }); } // namespace -using CLGEMMFixture = GEMMFixture<CLTensor, CLGEMM>; +using CLGEMMFixture = GEMMFixture<CLTensor, CLGEMM, CLAccessor>; TEST_SUITE(CL) diff --git a/tests/benchmark_new/NEON/GEMM.cpp b/tests/benchmark_new/NEON/GEMM.cpp index ad288267af..59ac4329c0 100644 --- a/tests/benchmark_new/NEON/GEMM.cpp +++ b/tests/benchmark_new/NEON/GEMM.cpp @@ -28,6 +28,7 @@ #include "arm_compute/runtime/TensorAllocator.h" #include "framework/Macros.h" #include "framework/datasets/Datasets.h" +#include "tests/NEON/Accessor.h" #include "tests/TypePrinter.h" #include "tests/datasets_new/MatrixMultiplyGEMMDataset.h" #include "tests/datasets_new/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1GEMMDataset.h" @@ -49,7 +50,7 @@ const auto data_types = framework::dataset::make("DataType", }); } // namespace -using NEGEMMFixture = GEMMFixture<Tensor, NEGEMM>; +using NEGEMMFixture = GEMMFixture<Tensor, NEGEMM, Accessor>; TEST_SUITE(NEON) diff --git a/tests/fixtures_new/GEMMFixture.h b/tests/fixtures_new/GEMMFixture.h index b23661f3e3..669a91f0cb 100644 --- a/tests/fixtures_new/GEMMFixture.h +++ b/tests/fixtures_new/GEMMFixture.h @@ -35,7 +35,7 @@ namespace arm_compute namespace test { /** Fixture that can be used for NEON and CL */ -template <typename TensorType, typename Function> +template <typename TensorType, typename Function, typename Accessor> class GEMMFixture : public framework::Fixture { public: @@ -58,6 +58,10 @@ public: b.allocator()->allocate(); c.allocator()->allocate(); dst.allocator()->allocate(); + + library->fill_tensor_uniform(Accessor(a), 0); + library->fill_tensor_uniform(Accessor(b), 1); + library->fill_tensor_uniform(Accessor(c), 2); } void run() |