diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-09-03 11:51:45 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 0d2b48c4a2cc82fd3312635a97117553ea4ee735 (patch) | |
| tree | 5b7ec8a0952fd0fe0d3fe598add9ca64fb03f473 /tests/benchmark | |
| parent | 565bf2d88df1790373f0ff92b8cc9e90dbf28d11 (diff) | |
| download | ComputeLibrary-0d2b48c4a2cc82fd3312635a97117553ea4ee735.tar.gz | |
COMPMID-1451: Add FP16 on NEON DragonBench
Change-Id: I245de06f93ea63458ff3e3e71ac8ec5737b80326
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/146614
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'tests/benchmark')
| -rw-r--r-- | tests/benchmark/NEON/DragonBench.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
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 |