616dd9ae14
This patch implements omp_get_device_num() in the host and the device. It uses the already existing getDeviceNum in the device config for the device. And in the host it uses the omp_get_num_devices(). Two simple tests added Differential Revision: https://reviews.llvm.org/D128347
51 lines
1.4 KiB
C++
51 lines
1.4 KiB
C++
//===--- Configuration.h - OpenMP device configuration interface -- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// API to query the global (constant) device environment.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef OMPTARGET_CONFIGURATION_H
|
|
#define OMPTARGET_CONFIGURATION_H
|
|
|
|
#include "Types.h"
|
|
|
|
namespace _OMP {
|
|
namespace config {
|
|
|
|
enum DebugKind : uint32_t {
|
|
Assertion = 1U << 0,
|
|
FunctionTracing = 1U << 1,
|
|
CommonIssues = 1U << 2,
|
|
};
|
|
|
|
/// Return the number of devices in the system, same number as returned on the
|
|
/// host by omp_get_num_devices.
|
|
uint32_t getNumDevices();
|
|
|
|
/// Return the device number in the system for omp_get_device_num.
|
|
uint32_t getDeviceNum();
|
|
|
|
/// Return the user choosen debug level.
|
|
uint32_t getDebugKind();
|
|
|
|
/// Return the amount of dynamic shared memory that was allocated at launch.
|
|
uint64_t getDynamicMemorySize();
|
|
|
|
/// Return if debugging is enabled for the given debug kind.
|
|
bool isDebugMode(DebugKind Level);
|
|
|
|
/// Indicates if this kernel may require thread-specific states, or if it was
|
|
/// explicitly disabled by the user.
|
|
bool mayUseThreadStates();
|
|
|
|
} // namespace config
|
|
} // namespace _OMP
|
|
|
|
#endif
|