openmp/libomptarget/DeviceRTL/src/Configuration.cpp

   1 //===- Configuration.cpp - OpenMP device configuration interface -- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file contains the data object of the constant device environment and the
  10 // query API.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "Configuration.h"
  15 #include "DeviceEnvironment.h"
  16 #include "State.h"
  17 #include "Types.h"
  18
  19 using namespace _OMP;
  20
  21 #pragma omp declare target
  22
  23 extern uint32_t __omp_rtl_debug_kind; // defined by CGOpenMPRuntimeGPU
  24
  25 // TODO: We want to change the name as soon as the old runtime is gone.
  26 DeviceEnvironmentTy CONSTANT(omptarget_device_environment)
  27     __attribute__((used, retain, weak));
  28
  29 uint32_t config::getDebugKind() {
  30   return __omp_rtl_debug_kind & omptarget_device_environment.DebugKind;
  31 }
  32
  33 uint32_t config::getNumDevices() {
  34   return omptarget_device_environment.NumDevices;
  35 }
  36
  37 uint32_t config::getDeviceNum() {
  38   return omptarget_device_environment.DeviceNum;
  39 }
  40
  41 uint64_t config::getDynamicMemorySize() {
  42   return omptarget_device_environment.DynamicMemSize;
  43 }
  44
  45 bool config::isDebugMode(config::DebugKind Kind) {
  46   return config::getDebugKind() & Kind;
  47 }
  48
  49 #pragma omp end declare target