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));
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
50