aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h')
-rw-r--r--tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h9
1 files changed, 5 insertions, 4 deletions
diff --git a/tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h b/tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h
index 70bafcc143..a9d6c9b6aa 100644
--- a/tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h
+++ b/tests/validation/fixtures/GEMMReshapeLHSMatrixFixture.h
@@ -46,7 +46,7 @@ namespace validation
{
using namespace arm_compute::misc::shape_calculator;
-template <typename TensorType, typename AccessorType, typename FunctionType, typename T, bool reinterpret_input_as_3d = false>
+template <typename TensorType, typename AccessorType, typename OperatorType, typename T, bool reinterpret_input_as_3d = false>
class GEMMReshapeLHSMatrixValidationFixture : public framework::Fixture
{
public:
@@ -86,8 +86,8 @@ protected:
// The output tensor will be auto-initialized within the function
// Create and configure function
- FunctionType gemm_lhs_reshape;
- gemm_lhs_reshape.configure(&src, &dst, lhs_info, reinterpret_input_as_3d);
+ OperatorType gemm_lhs_reshape;
+ gemm_lhs_reshape.configure(src.info(), dst.info(), lhs_info, reinterpret_input_as_3d);
ARM_COMPUTE_ASSERT(src.info()->is_resizable());
@@ -104,7 +104,8 @@ protected:
fill(AccessorType(src));
// Compute GEMM LHS matrix reshape function
- gemm_lhs_reshape.run();
+ ITensorPack tensors = { { ACL_SRC, &src }, { ACL_DST, &dst } };
+ gemm_lhs_reshape.run(tensors);
return dst;
}