tests/NEON/Helper.h

   1 /*
   2  * Copyright (c) 2017-2018 ARM Limited.
   3  *
   4  * SPDX-License-Identifier: MIT
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to
   8  * deal in the Software without restriction, including without limitation the
   9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  10  * sell copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in all
  14  * copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22  * SOFTWARE.
  23  */
  24 #ifndef __ARM_COMPUTE_TEST_NEON_HELPER_H__
  25 #define __ARM_COMPUTE_TEST_NEON_HELPER_H__
  26
  27 #include "arm_compute/runtime/Array.h"
  28 #include "arm_compute/runtime/NEON/INESimpleFunction.h"
  29 #include "support/ToolchainSupport.h"
  30 #include "tests/Globals.h"
  31
  32 #include <algorithm>
  33 #include <array>
  34 #include <vector>
  35
  36 namespace arm_compute
  37 {
  38 namespace test
  39 {
  40 template <typename D, typename T, typename... Ts>
  41 void fill_tensors(D &&dist, std::initializer_list<int> seeds, T &&tensor, Ts &&... other_tensors)
  42 {
  43     const std::array < T, 1 + sizeof...(Ts) > tensors{ { std::forward<T>(tensor), std::forward<Ts>(other_tensors)... } };
  44     std::vector<int> vs(seeds);
  45     ARM_COMPUTE_ERROR_ON(vs.size() != tensors.size());
  46     int k = 0;
  47     for(auto tp : tensors)
  48     {
  49         library->fill(Accessor(*tp), std::forward<D>(dist), vs[k++]);
  50     }
  51 }
  52
  53 /** This template synthetizes an INESimpleFunction which runs the given kernel K */
  54 template <typename K>
  55 class NESynthetizeFunction : public INESimpleFunction
  56 {
  57 public:
  58     /** Configure the kernel.
  59      *
  60      * @param[in] args Configuration arguments.
  61      */
  62     template <typename... Args>
  63     void configure(Args &&... args)
  64     {
  65         auto k = arm_compute::support::cpp14::make_unique<K>();
  66         k->configure(std::forward<Args>(args)...);
  67         _kernel = std::move(k);
  68     }
  69 };
  70
  71 /** As above but this also setups a Zero border on the input tensor of the specified bordersize */
  72 template <typename K, int bordersize>
  73 class NESynthetizeFunctionWithZeroConstantBorder : public INESimpleFunction
  74 {
  75 public:
  76     /** Configure the kernel.
  77      *
  78      * @param[in] first First configuration argument.
  79      * @param[in] args  Rest of the configuration arguments.
  80      */
  81     template <typename T, typename... Args>
  82     void configure(T first, Args &&... args)
  83     {
  84         auto k = arm_compute::support::cpp14::make_unique<K>();
  85         k->configure(first, std::forward<Args>(args)...);
  86         _kernel = std::move(k);
  87         _border_handler.configure(first, BorderSize(bordersize), BorderMode::CONSTANT, PixelValue(0));
  88     }
  89 };
  90
  91 } // namespace test
  92 } // namespace arm_compute
  93 #endif /* __ARM_COMPUTE_TEST_NEON_HELPER_H__ */