460840c09d
Summary: The `omp_get_num_procs()` function should return the amount of parallelism availible. On the GPU, this was not defined. We have elected to define this function as the maximum amount of wavefronts / warps that can be simultaneously resident on the device. For AMDGPU this is the number of CUs multiplied byth CU's per wave. For NVPTX this is the maximum threads per SM divided by the warp size and multiplied by the number of SMs.
67 lines
1.9 KiB
C++
67 lines
1.9 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 ompx {
|
|
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();
|
|
|
|
/// Returns the cycles per second of the device's fixed frequency clock.
|
|
uint64_t getClockFrequency();
|
|
|
|
/// Returns the pointer to the beginning of the indirect call table.
|
|
void *getIndirectCallTablePtr();
|
|
|
|
/// Returns the size of the indirect call table.
|
|
uint64_t getIndirectCallTableSize();
|
|
|
|
/// Returns the size of the indirect call table.
|
|
uint64_t getHardwareParallelism();
|
|
|
|
/// 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();
|
|
|
|
/// Indicates if this kernel may require data environments for nested
|
|
/// parallelism, or if it was explicitly disabled by the user.
|
|
bool mayUseNestedParallelism();
|
|
|
|
} // namespace config
|
|
} // namespace ompx
|
|
|
|
#endif
|