inference-engine/src/mkldnn_plugin/mkldnn/omp_manager.h

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 /**
   6  * @brief Abstraction over platform specific implementations
   7  * @file omp_manager.h
   8  */
   9 #pragma once
  10
  11 namespace MKLDNNPlugin {
  12 namespace cpu {
  13
  14 bool checkOpenMpEnvVars(bool includeOMPNumThreads = true);
  15 // numbers of CPU sockets in the machine (on Linux), 1 on all other OSes
  16 int getNumberOfCPUSockets();
  17 // numbers of CPU physical cores on Linux (which is considered to be more performance friendly for servers)
  18 // (on other OSes it simply relies on the original parallel API of choice, which usually use the logical cores )
  19 int getNumberOfCPUCores();
  20
  21 }  // namespace cpu
  22 }  // namespace MKLDNNPlugin