diff options
Diffstat (limited to 'tests/validation/fixtures/GEMMFixture.h')
-rw-r--r-- | tests/validation/fixtures/GEMMFixture.h | 30 |
1 files changed, 14 insertions, 16 deletions
diff --git a/tests/validation/fixtures/GEMMFixture.h b/tests/validation/fixtures/GEMMFixture.h index 8dd2998377..e4762cc5be 100644 --- a/tests/validation/fixtures/GEMMFixture.h +++ b/tests/validation/fixtures/GEMMFixture.h @@ -47,13 +47,12 @@ class GEMMValidationFixedPointFixture : public framework::Fixture { public: template <typename...> - void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape, float alpha, float beta, DataType data_type, int fractional_bits) + void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape, float alpha, float beta, DataType data_type) { - _fractional_bits = fractional_bits; - _data_type = data_type; + _data_type = data_type; - _target = compute_target(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type, fractional_bits); - _reference = compute_reference(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type, fractional_bits); + _target = compute_target(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type); + _reference = compute_reference(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type); } protected: @@ -75,13 +74,13 @@ protected: } TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_c, const TensorShape &output_shape, float alpha, float beta, - DataType data_type, int fixed_point_position) + DataType data_type) { // Create tensors - TensorType a = create_tensor<TensorType>(shape_a, data_type, 1, fixed_point_position); - TensorType b = create_tensor<TensorType>(shape_b, data_type, 1, fixed_point_position); - TensorType c = create_tensor<TensorType>(shape_c, data_type, 1, fixed_point_position); - TensorType dst = create_tensor<TensorType>(output_shape, data_type, 1, fixed_point_position); + TensorType a = create_tensor<TensorType>(shape_a, data_type, 1); + TensorType b = create_tensor<TensorType>(shape_b, data_type, 1); + TensorType c = create_tensor<TensorType>(shape_c, data_type, 1); + TensorType dst = create_tensor<TensorType>(output_shape, data_type, 1); // Create and configure function FunctionType gemm; @@ -120,12 +119,12 @@ protected: } SimpleTensor<T> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShape &shape_c, const TensorShape &output_shape, float alpha, float beta, - DataType data_type, int fixed_point_position) + DataType data_type) { // Create reference - SimpleTensor<T> a{ shape_a, data_type, 1, fixed_point_position }; - SimpleTensor<T> b{ shape_b, data_type, 1, fixed_point_position }; - SimpleTensor<T> c{ shape_c, data_type, 1, fixed_point_position }; + SimpleTensor<T> a{ shape_a, data_type, 1 }; + SimpleTensor<T> b{ shape_b, data_type, 1 }; + SimpleTensor<T> c{ shape_c, data_type, 1 }; // Fill reference fill(a, 0); @@ -137,7 +136,6 @@ protected: TensorType _target{}; SimpleTensor<T> _reference{}; - int _fractional_bits{}; DataType _data_type{}; }; @@ -148,7 +146,7 @@ public: template <typename...> void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape, float alpha, float beta, DataType data_type) { - GEMMValidationFixedPointFixture<TensorType, AccessorType, FunctionType, T>::setup(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type, 0); + GEMMValidationFixedPointFixture<TensorType, AccessorType, FunctionType, T>::setup(shape_a, shape_b, shape_c, output_shape, alpha, beta, data_type); } }; } // namespace validation |