08533a3ee8
We had three `utils::` namespaces, all with different "meaning" (host, device, hsa_utils). We should, when we can, keep "include/Shared" accessible from host and device, thus RefCountTy has been moved to a separate header. `hsa_utils` was introduced to make `utils::` less overloaded. And common functionality was de-duplicated, e.g., `utils::advance` and `utils::advanceVoidPtr` -> `utils:advancePtr`. Type punning now checks for the size of the result to make sure it matches the source type. No functional change was intended.
90 lines
2.8 KiB
C++
90 lines
2.8 KiB
C++
//===- Configuration.cpp - 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains the data object of the constant device environment and the
|
|
// query API.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Configuration.h"
|
|
#include "DeviceTypes.h"
|
|
#include "State.h"
|
|
|
|
using namespace ompx;
|
|
|
|
#pragma omp begin declare target device_type(nohost)
|
|
|
|
// Weak definitions will be overridden by CGOpenmpRuntimeGPU if enabled.
|
|
[[gnu::weak]] extern const uint32_t __omp_rtl_debug_kind = 0;
|
|
[[gnu::weak]] extern const uint32_t __omp_rtl_assume_no_thread_state = 0;
|
|
[[gnu::weak]] extern const uint32_t __omp_rtl_assume_no_nested_parallelism = 0;
|
|
[[gnu::weak]] extern const uint32_t __omp_rtl_assume_threads_oversubscription =
|
|
0;
|
|
[[gnu::weak]] extern const uint32_t __omp_rtl_assume_teams_oversubscription = 0;
|
|
|
|
// This variable should be visibile to the plugin so we override the default
|
|
// hidden visibility.
|
|
[[gnu::used, gnu::retain, gnu::weak,
|
|
gnu::visibility("protected")]] DeviceEnvironmentTy
|
|
CONSTANT(__omp_rtl_device_environment);
|
|
|
|
uint32_t config::getAssumeTeamsOversubscription() {
|
|
return __omp_rtl_assume_teams_oversubscription;
|
|
}
|
|
|
|
uint32_t config::getAssumeThreadsOversubscription() {
|
|
return __omp_rtl_assume_threads_oversubscription;
|
|
}
|
|
|
|
uint32_t config::getDebugKind() {
|
|
return __omp_rtl_debug_kind & __omp_rtl_device_environment.DeviceDebugKind;
|
|
}
|
|
|
|
uint32_t config::getNumDevices() {
|
|
return __omp_rtl_device_environment.NumDevices;
|
|
}
|
|
|
|
uint32_t config::getDeviceNum() {
|
|
return __omp_rtl_device_environment.DeviceNum;
|
|
}
|
|
|
|
uint64_t config::getDynamicMemorySize() {
|
|
return __omp_rtl_device_environment.DynamicMemSize;
|
|
}
|
|
|
|
uint64_t config::getClockFrequency() {
|
|
return __omp_rtl_device_environment.ClockFrequency;
|
|
}
|
|
|
|
void *config::getIndirectCallTablePtr() {
|
|
return reinterpret_cast<void *>(
|
|
__omp_rtl_device_environment.IndirectCallTable);
|
|
}
|
|
|
|
uint64_t config::getHardwareParallelism() {
|
|
return __omp_rtl_device_environment.HardwareParallelism;
|
|
}
|
|
|
|
uint64_t config::getIndirectCallTableSize() {
|
|
return __omp_rtl_device_environment.IndirectCallTableSize;
|
|
}
|
|
|
|
bool config::isDebugMode(DeviceDebugKind Kind) {
|
|
return config::getDebugKind() & uint32_t(Kind);
|
|
}
|
|
|
|
bool config::mayUseThreadStates() { return !__omp_rtl_assume_no_thread_state; }
|
|
|
|
bool config::mayUseNestedParallelism() {
|
|
if (__omp_rtl_assume_no_nested_parallelism)
|
|
return false;
|
|
return state::getKernelEnvironment().Configuration.MayUseNestedParallelism;
|
|
}
|
|
|
|
#pragma omp end declare target
|