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 // This variable should be visibile to the plugin so we override the default
27 // hidden visibility.
28 DeviceEnvironmentTy CONSTANT(omptarget_device_environment)
29     __attribute__((used, retain, weak, visibility("protected")));
30 
31 uint32_t config::getDebugKind() {
32   return __omp_rtl_debug_kind & omptarget_device_environment.DebugKind;
33 }
34 
35 uint32_t config::getNumDevices() {
36   return omptarget_device_environment.NumDevices;
37 }
38 
39 uint32_t config::getDeviceNum() {
40   return omptarget_device_environment.DeviceNum;
41 }
42 
43 uint64_t config::getDynamicMemorySize() {
44   return omptarget_device_environment.DynamicMemSize;
45 }
46 
47 bool config::isDebugMode(config::DebugKind Kind) {
48   return config::getDebugKind() & Kind;
49 }
50 
51 #pragma omp end declare target
52