src/graph/GraphContext.cpp

   1 /*
   2  * Copyright (c) 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 #include "arm_compute/graph/GraphContext.h"
  25 #include <arm_compute/graph.h>
  26
  27 namespace arm_compute
  28 {
  29 namespace graph
  30 {
  31 GraphContext::GraphContext()
  32     : _config(), _memory_managers()
  33 {
  34 }
  35
  36 const GraphConfig &GraphContext::config() const
  37 {
  38     return _config;
  39 }
  40
  41 void GraphContext::set_config(const GraphConfig &config)
  42 {
  43     _config = config;
  44 }
  45
  46 bool GraphContext::insert_memory_management_ctx(MemoryManagerContext &&memory_ctx)
  47 {
  48     Target target = memory_ctx.target;
  49     if(target == Target::UNSPECIFIED || _memory_managers.find(target) != std::end(_memory_managers))
  50     {
  51         return false;
  52     }
  53
  54     _memory_managers[target] = std::move(memory_ctx);
  55     return true;
  56 }
  57
  58 MemoryManagerContext *GraphContext::memory_management_ctx(Target target)
  59 {
  60     return (_memory_managers.find(target) != std::end(_memory_managers)) ? &_memory_managers[target] : nullptr;
  61 }
  62
  63 std::map<Target, MemoryManagerContext> &GraphContext::memory_managers()
  64 {
  65     return _memory_managers;
  66 }
  67
  68 void GraphContext::finalize()
  69 {
  70     for(auto &mm_obj : _memory_managers)
  71     {
  72         // Finalize intra layer memory manager
  73         if(mm_obj.second.intra_mm != nullptr)
  74         {
  75             mm_obj.second.intra_mm->finalize();
  76         }
  77         // Finalize cross layer memory manager
  78         if(mm_obj.second.cross_mm != nullptr)
  79         {
  80             mm_obj.second.cross_mm->finalize();
  81         }
  82     }
  83 }
  84 } // namespace graph
  85 } // namespace arm_compute