runtime/onert/backend/ruy/ExternalContext.h

   1 /*
   2  * Copyright (c) 2020 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 #ifndef __ONERT_BACKEND_RUY_EXTERNAL_CONTEXT_H__
  18 #define __ONERT_BACKEND_RUY_EXTERNAL_CONTEXT_H__
  19
  20 #include <util/ConfigSource.h>
  21 #include <ruy/context.h>
  22
  23 namespace
  24 {
  25 const int kDefaultNumThreadpoolThreads = 4;
  26 }
  27
  28 namespace onert
  29 {
  30 namespace backend
  31 {
  32 namespace ruy
  33 {
  34
  35 class ExternalContext
  36 {
  37 public:
  38   ExternalContext() : _ruy_context(new ::ruy::Context)
  39   {
  40     setMaxNumThreads(onert::util::getConfigInt(onert::util::config::RUY_THREADS));
  41   }
  42
  43   void setMaxNumThreads(int max_num_threads)
  44   {
  45     const int target_num_threads =
  46         max_num_threads > -1 ? max_num_threads : kDefaultNumThreadpoolThreads;
  47     _ruy_context->set_max_num_threads(target_num_threads);
  48   }
  49
  50   ::ruy::Context *ruy_context() const { return _ruy_context.get(); }
  51
  52 private:
  53   const std::unique_ptr<::ruy::Context> _ruy_context;
  54 };
  55
  56 } // namespace ruy
  57 } // namespace backend
  58 } // namespace onert
  59
  60 #endif // __ONERT_BACKEND_RUY_EXTERNAL_CONTEXT_H__