runtime/onert/core/include/util/Utils.h

   1 /*
   2  * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *    http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /**
  18  * @file     Utils.h
  19  * @brief    This file contains utility macro
  20  */
  21
  22 #ifndef __ONERT_UTIL_UTILS_H__
  23 #define __ONERT_UTIL_UTILS_H__
  24
  25 #include "ir/Coordinates.h"
  26 #include "ir/Shape.h"
  27
  28 #define UNUSED_RELEASE(a) (void)(a)
  29
  30 template <size_t rest> struct ForEachDimension
  31 {
  32   template <typename L>
  33   static void unroll(const onert::ir::Shape &shape, onert::ir::Coordinates &coords,
  34                      L lambda_function)
  35   {
  36     if (static_cast<int>(rest) > shape.rank())
  37     {
  38       ForEachDimension<rest - 1>::unroll(shape, coords, lambda_function);
  39       return;
  40     }
  41
  42     const auto axis = shape.rank() - rest;
  43     const auto &d = shape.dim(axis);
  44
  45     for (auto v = 0; v < d; v++)
  46     {
  47       coords.set(axis, v);
  48       ForEachDimension<rest - 1>::unroll(shape, coords, lambda_function);
  49     }
  50   }
  51 };
  52
  53 template <> struct ForEachDimension<0>
  54 {
  55   template <typename L>
  56   static void unroll(const onert::ir::Shape &shape, onert::ir::Coordinates &coords,
  57                      L lambda_function)
  58   {
  59     UNUSED_RELEASE(shape);
  60     lambda_function(coords);
  61   }
  62 };
  63
  64 template <typename L> inline void ShapeLoop(const onert::ir::Shape &shape, L lambda_function)
  65 {
  66   int32_t rank = shape.rank();
  67   assert(rank > 0);
  68   for (int32_t i = 0; i < rank; ++i)
  69   {
  70     assert(shape.dim(i) > 0);
  71   }
  72
  73   onert::ir::Coordinates coords;
  74   if (rank == 0)
  75   {
  76     coords.set(0, 0);
  77   }
  78   // TODO Change 6 to onert::ir::Shape::kMaxRank if onert::ir::Shape::kMaxRank is modified as a
  79   // constant expression
  80   ForEachDimension<6>::unroll(shape, coords, lambda_function);
  81 }
  82 #endif // __ONERT_UTIL_UTILS_H__