diff options
Diffstat (limited to 'tests/validate_examples/cl_gemm.cpp')
-rw-r--r-- | tests/validate_examples/cl_gemm.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/validate_examples/cl_gemm.cpp b/tests/validate_examples/cl_gemm.cpp index 39fe111448..cdf60cd65b 100644 --- a/tests/validate_examples/cl_gemm.cpp +++ b/tests/validate_examples/cl_gemm.cpp @@ -195,11 +195,13 @@ public: consume_params(gemm_options); print_parameters_internal(); + const bool is_quantized = is_data_type_quantized(data_type); + // Calculate re-quantization parameters - if(data_type == DataType::QASYMM8) + if(is_quantized) { float multiplier = scale_src0 * scale_src1 / scale_dst; - quantization::calculate_quantized_multiplier_less_than_one(multiplier, &dst_multiplier, &dst_shift); + quantization::calculate_quantized_multiplier(multiplier, &dst_multiplier, &dst_shift); } // Initialize GEMM inputs/outputs @@ -209,7 +211,7 @@ public: init_sgemm_output(dst, src0, src1, data_type); // Configure function - if(data_type == DataType::QASYMM8) + if(is_quantized) { src0.info()->set_quantization_info(QuantizationInfo(scale_src0, offset_src0)); src1.info()->set_quantization_info(QuantizationInfo(scale_src1, offset_src1)); |