6dd791bca8
A common problem is the device running out of global heap memory and crashing due to a nullptr dereference when using the data sharing stack. This explicitly checks that a nullptr was not returned by malloc when debugging field 1 is enabled. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D112005
47 lines
1.3 KiB
C++
47 lines
1.3 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 number of devices in the system, same number as returned on the
|
|
/// host by omp_get_num_devices.
|
|
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();
|
|
|
|
bool isDebugMode(DebugKind Level);
|
|
|
|
} // namespace config
|
|
} // namespace _OMP
|
|
|
|
#endif
|