diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/benchmark/CL/DragonBench.cpp | 2 | ||||
-rw-r--r-- | tests/benchmark/NEON/DragonBench.cpp | 4 | ||||
-rw-r--r-- | tests/benchmark/fixtures/DragonBenchFixture.h | 11 |
3 files changed, 9 insertions, 8 deletions
diff --git a/tests/benchmark/CL/DragonBench.cpp b/tests/benchmark/CL/DragonBench.cpp index 6e047a5fa1..d83878fa21 100644 --- a/tests/benchmark/CL/DragonBench.cpp +++ b/tests/benchmark/CL/DragonBench.cpp @@ -47,7 +47,7 @@ namespace benchmark namespace { // Common DragonBench parameters -auto CommonParams = combine(combine(framework::dataset::make("DataType", { DataType::F16, DataType::F32 }), +auto CommonParams = combine(combine(framework::dataset::make("DataType", { DataType::QASYMM8, DataType::F16, DataType::F32 }), framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })), framework::dataset::make("HasBias", { true, false })); } // namespace diff --git a/tests/benchmark/NEON/DragonBench.cpp b/tests/benchmark/NEON/DragonBench.cpp index 5544b8a9e7..31bf6efb70 100644 --- a/tests/benchmark/NEON/DragonBench.cpp +++ b/tests/benchmark/NEON/DragonBench.cpp @@ -47,9 +47,9 @@ namespace benchmark namespace { #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC -const auto data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32 }); +const auto data_types = framework::dataset::make("DataType", { DataType::QASYMM8, DataType::F16, DataType::F32 }); #else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ -const auto data_types = framework::dataset::make("DataType", { DataType::F32 }); +const auto data_types = framework::dataset::make("DataType", { DataType::QASYMM8, DataType::F32 }); #endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ // Common DragonBench parameters diff --git a/tests/benchmark/fixtures/DragonBenchFixture.h b/tests/benchmark/fixtures/DragonBenchFixture.h index b9b3a180ac..526a798154 100644 --- a/tests/benchmark/fixtures/DragonBenchFixture.h +++ b/tests/benchmark/fixtures/DragonBenchFixture.h @@ -68,13 +68,14 @@ public: } // Determine bias data type - DataType bias_data_type = is_data_type_quantized_asymmetric(data_type) ? DataType::S32 : data_type; + DataType bias_data_type = is_data_type_quantized_asymmetric(data_type) ? DataType::S32 : data_type; + const QuantizationInfo q_info(2.f, 10); // Create tensors - src = create_tensor<TensorType>(src_shape, data_type, 1, QuantizationInfo(), data_layout); - weights = create_tensor<TensorType>(weights_shape, data_type, 1, QuantizationInfo(), data_layout); - biases = create_tensor<TensorType>(biases_shape, bias_data_type, 1, QuantizationInfo(), data_layout); - dst = create_tensor<TensorType>(dst_shape, data_type, 1, QuantizationInfo(), data_layout); + src = create_tensor<TensorType>(src_shape, data_type, 1, q_info, data_layout); + weights = create_tensor<TensorType>(weights_shape, data_type, 1, q_info, data_layout); + biases = create_tensor<TensorType>(biases_shape, bias_data_type, 1, q_info, data_layout); + dst = create_tensor<TensorType>(dst_shape, data_type, 1, q_info, data_layout); // Create and configure function conv_layer.configure(&src, &weights, has_bias ? &biases : nullptr, &dst, info); |