From fa4cacdff825a38eac31ef7ecd3ad6b30da53eaa Mon Sep 17 00:00:00 2001 From: Gian Marco Date: Wed, 18 Oct 2017 17:05:02 +0100 Subject: COMPMID-636 - Extending GEMMLowp validation for NEON intrinsics Change-Id: Id02a0b3bf5af65dd940b46b2f6634b6a479cf388 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/92275 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- tests/datasets/GEMMLowpDataset.h | 147 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 147 insertions(+) create mode 100644 tests/datasets/GEMMLowpDataset.h (limited to 'tests/datasets/GEMMLowpDataset.h') diff --git a/tests/datasets/GEMMLowpDataset.h b/tests/datasets/GEMMLowpDataset.h new file mode 100644 index 0000000000..4bf2a98d61 --- /dev/null +++ b/tests/datasets/GEMMLowpDataset.h @@ -0,0 +1,147 @@ +/* + * Copyright (c) 2017 ARM Limited. + * + * SPDX-License-Identifier: MIT + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to + * deal in the Software without restriction, including without limitation the + * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or + * sell copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +#ifndef ARM_COMPUTE_TEST_GEMMLOWP_DATASET +#define ARM_COMPUTE_TEST_GEMMLOWP_DATASET + +#include "utils/TypePrinter.h" + +#include "arm_compute/core/TensorShape.h" + +namespace arm_compute +{ +namespace test +{ +namespace datasets +{ +class GEMMLowpDataset +{ +public: + using type = std::tuple; + + struct iterator + { + iterator(std::vector::const_iterator a_it, + std::vector::const_iterator b_it, + std::vector::const_iterator c_it, + std::vector::const_iterator a_offset_it, + std::vector::const_iterator b_offset_it, + std::vector::const_iterator c_offset_it, + std::vector::const_iterator c_mult_int_it, + std::vector::const_iterator out_shift_it) + : _a_it{ std::move(a_it) }, + _b_it{ std::move(b_it) }, + _c_it{ std::move(c_it) }, + _a_offset_it{ std::move(a_offset_it) }, + _b_offset_it{ std::move(b_offset_it) }, + _c_offset_it{ std::move(c_offset_it) }, + _c_mult_int_it{ std::move(c_mult_int_it) }, + _out_shift_it{ std::move(out_shift_it) } + { + } + + std::string description() const + { + std::stringstream description; + description << "A=" << *_a_it << ":"; + description << "B=" << *_b_it << ":"; + description << "C=" << *_c_it << ":"; + description << "a_offset=" << *_a_offset_it << ":"; + description << "b_offset=" << *_b_offset_it << ":"; + description << "c_offset=" << *_c_offset_it << ":"; + description << "c_mult_int=" << *_c_mult_int_it << ":"; + description << "out_shift=" << *_out_shift_it << ":"; + return description.str(); + } + + GEMMLowpDataset::type operator*() const + { + return std::make_tuple(*_a_it, *_b_it, *_c_it, *_a_offset_it, *_b_offset_it, *_c_offset_it, *_c_mult_int_it, *_out_shift_it); + } + + iterator &operator++() + { + ++_a_it; + ++_b_it; + ++_c_it; + ++_a_offset_it; + ++_b_offset_it; + ++_c_offset_it; + ++_c_mult_int_it; + ++_out_shift_it; + + return *this; + } + + private: + std::vector::const_iterator _a_it; + std::vector::const_iterator _b_it; + std::vector::const_iterator _c_it; + std::vector::const_iterator _a_offset_it; + std::vector::const_iterator _b_offset_it; + std::vector::const_iterator _c_offset_it; + std::vector::const_iterator _c_mult_int_it; + std::vector::const_iterator _out_shift_it; + }; + + iterator begin() const + { + return iterator(_a_shapes.begin(), _b_shapes.begin(), _c_shapes.begin(), _a_offset.begin(), _b_offset.begin(), _c_offset.begin(), _c_mult_int.begin(), _out_shift.begin()); + } + + int size() const + { + return std::min(_a_shapes.size(), std::min(_b_shapes.size(), std::min(_c_shapes.size(), std::min(_a_offset.size(), std::min(_b_offset.size(), std::min(_c_offset.size(), std::min(_c_mult_int.size(), + _out_shift.size()))))))); + } + + void add_config(TensorShape a, TensorShape b, TensorShape c, int32_t a_offset, int32_t b_offset, int32_t c_offset, int32_t c_mult_int, int32_t out_shift) + { + _a_shapes.emplace_back(std::move(a)); + _b_shapes.emplace_back(std::move(b)); + _c_shapes.emplace_back(std::move(c)); + _a_offset.emplace_back(std::move(a_offset)); + _b_offset.emplace_back(std::move(b_offset)); + _c_offset.emplace_back(std::move(c_offset)); + _c_mult_int.emplace_back(std::move(c_mult_int)); + _out_shift.emplace_back(std::move(out_shift)); + } + +protected: + GEMMLowpDataset() = default; + GEMMLowpDataset(GEMMLowpDataset &&) = default; + +private: + std::vector _a_shapes{}; + std::vector _b_shapes{}; + std::vector _c_shapes{}; + std::vector _a_offset{}; + std::vector _b_offset{}; + std::vector _c_offset{}; + std::vector _c_mult_int{}; + std::vector _out_shift{}; +}; +} // namespace datasets +} // namespace test +} // namespace arm_compute +#endif /* ARM_COMPUTE_TEST_GEMMLOWP_DATASET */ -- cgit v1.2.1