From 0d2b48c4a2cc82fd3312635a97117553ea4ee735 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 3 Sep 2018 11:51:45 +0100 Subject: COMPMID-1451: Add FP16 on NEON DragonBench Change-Id: I245de06f93ea63458ff3e3e71ac8ec5737b80326 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/146614 Tested-by: Jenkins Reviewed-by: Michalis Spyrou --- tests/benchmark/NEON/DragonBench.cpp | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'tests/benchmark') diff --git a/tests/benchmark/NEON/DragonBench.cpp b/tests/benchmark/NEON/DragonBench.cpp index a1d826d343..5544b8a9e7 100644 --- a/tests/benchmark/NEON/DragonBench.cpp +++ b/tests/benchmark/NEON/DragonBench.cpp @@ -46,9 +46,14 @@ namespace benchmark { namespace { +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +const auto data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32 }); +#else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ +const auto data_types = framework::dataset::make("DataType", { DataType::F32 }); +#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ + // Common DragonBench parameters -auto CommonParams = combine(combine(framework::dataset::make("DataType", { DataType::F32 }), - framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })), +auto CommonParams = combine(combine(data_types, framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })), framework::dataset::make("HasBias", { true, false })); } // namespace -- cgit v1.2.1