compiler/luci/pass/src/ReplaceSubWithAddPass.cpp

   1 /*
   2  * Copyright (c) 2021 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 #include "luci/Pass/ReplaceSubWithAddPass.h"
  18
  19 #include <luci/IR/CircleNodes.h>
  20 #include <luci/Profile/CircleNodeOrigin.h>
  21 #include <luci/Service/Nodes/CircleConst.h>
  22
  23 namespace
  24 {
  25
  26 bool replace_sub_with_const_rhs(luci::CircleSub *sub)
  27 {
  28   auto const_rhs = dynamic_cast<luci::CircleConst *>(sub->y());
  29   if (const_rhs == nullptr)
  30     return false;
  31
  32   auto graph = sub->graph();
  33
  34   auto neg_const_rhs = luci::clone(const_rhs);
  35   if (neg_const_rhs->dtype() == loco::DataType::FLOAT32)
  36   {
  37     for (uint32_t i = 0; i < neg_const_rhs->size<loco::DataType::FLOAT32>(); ++i)
  38       neg_const_rhs->at<loco::DataType::FLOAT32>(i) *= -1.0;
  39   }
  40   else
  41   {
  42     // TODO Support more data type
  43     return false;
  44   }
  45
  46   auto add = graph->nodes()->create<luci::CircleAdd>();
  47   add->x(sub->x());
  48   add->y(neg_const_rhs);
  49   add->name(sub->name());
  50   add->fusedActivationFunction(sub->fusedActivationFunction());
  51   luci::add_origin(add, luci::get_origin(sub));
  52   loco::replace(sub).with(add);
  53   return true;
  54 }
  55
  56 } // namespace
  57
  58 namespace luci
  59 {
  60
  61 bool ReplaceSubWithAddPass::run(loco::Graph *g)
  62 {
  63   bool changed = false;
  64   for (auto node : loco::active_nodes(loco::output_nodes(g)))
  65   {
  66     if (auto sub = dynamic_cast<luci::CircleSub *>(node))
  67     {
  68       if (replace_sub_with_const_rhs(sub))
  69         changed = true;
  70     }
  71   }
  72
  73   return changed;
  74 }
  75
  76 } // namespace luci