be6f110bc0
Summary:
Currently we build the OpenMP device runtime as part of the `offload/`
project. This is problematic because it has several restrictions when
compared to the normal offloading runtime. It can only be built with an
up-to-date clang and we need to set the target appropriately. Currently
we hack around this by creating the compiler invocation manually, but
this patch moves it into a separate runtimes build.
This follows the same build we use for libc, libc++, compiler-rt, and
flang-rt. This also moves it from `offload/` into `openmp/` because it
is still the `openmp/` runtime and I feel it is more appropriate. We do
want a generic `offload/` library at some point, but it would be trivial
to then add that as a separate library now that we have the
infrastructure that makes adding these new libraries trivial.
This most importantly will require that users update their build
configs, mostly adding the following lines at a minimum. I was debating
whether or not I should 'auto-upgrade' this, but I just went with a
warning.
```
-DLLVM_RUNTIME_TARGETS='default;amdgcn-amd-amdhsa;nvptx64-nvidia-cuda' \
-DRUNTIMES_nvptx64-nvidia-cuda_LLVM_ENABLE_RUNTIMES=openmp \
-DRUNTIMES_amdgcn-amd-amdhsa_LLVM_ENABLE_RUNTIMES=openmp \
```
This also changed where the `.bc` version of the library lives, but it's
still created.
109 lines
3.5 KiB
C++
109 lines
3.5 KiB
C++
//===--------- Mapping.h - OpenMP device runtime mapping helpers -- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef OMPTARGET_MAPPING_H
|
|
#define OMPTARGET_MAPPING_H
|
|
|
|
#include "DeviceTypes.h"
|
|
|
|
namespace ompx {
|
|
|
|
namespace mapping {
|
|
|
|
enum {
|
|
DIM_X = __GPU_X_DIM,
|
|
DIM_Y = __GPU_Y_DIM,
|
|
DIM_Z = __GPU_Z_DIM,
|
|
};
|
|
|
|
inline constexpr uint32_t MaxThreadsPerTeam = 1024;
|
|
|
|
/// Initialize the mapping machinery.
|
|
void init(bool IsSPMD);
|
|
|
|
/// Return true if the kernel is executed in SPMD mode.
|
|
bool isSPMDMode();
|
|
|
|
/// Return true if the kernel is executed in generic mode.
|
|
bool isGenericMode();
|
|
|
|
/// Return true if the executing thread is the main thread in generic mode.
|
|
/// These functions will lookup state and it is required that that is OK for the
|
|
/// thread and location. See also `isInitialThreadInLevel0` for a stateless
|
|
/// alternative for certain situations, e.g. during initialization.
|
|
bool isMainThreadInGenericMode();
|
|
bool isMainThreadInGenericMode(bool IsSPMD);
|
|
|
|
/// Return true if this thread is the initial thread in parallel level 0.
|
|
///
|
|
/// The thread for which this returns true should be used for single threaded
|
|
/// initialization tasks. We pick a special thread to ensure there are no
|
|
/// races between the initialization and the first read of initialized state.
|
|
bool isInitialThreadInLevel0(bool IsSPMD);
|
|
|
|
/// Return true if the executing thread has the lowest Id of the active threads
|
|
/// in the warp.
|
|
bool isLeaderInWarp();
|
|
|
|
/// Return a mask describing all active threads in the warp.
|
|
LaneMaskTy activemask();
|
|
|
|
/// Return a mask describing all threads with a smaller Id in the warp.
|
|
LaneMaskTy lanemaskLT();
|
|
|
|
/// Return a mask describing all threads with a larger Id in the warp.
|
|
LaneMaskTy lanemaskGT();
|
|
|
|
/// Return the thread Id in the warp, in [0, getWarpSize()).
|
|
uint32_t getThreadIdInWarp();
|
|
|
|
/// Return the warp size, thus number of threads in the warp.
|
|
uint32_t getWarpSize();
|
|
|
|
/// Return the warp id in the block, in [0, getNumberOfWarpsInBlock()]
|
|
uint32_t getWarpIdInBlock();
|
|
|
|
/// Return the number of warps in the block.
|
|
uint32_t getNumberOfWarpsInBlock();
|
|
|
|
/// Return the thread Id in the block, in [0, getNumberOfThreadsInBlock(Dim)).
|
|
uint32_t getThreadIdInBlock(int32_t Dim = DIM_X);
|
|
|
|
/// Return the block size, thus number of threads in the block.
|
|
uint32_t getNumberOfThreadsInBlock(int32_t Dim = DIM_X);
|
|
|
|
/// Return the block Id in the kernel, in [0, getNumberOfBlocksInKernel(Dim)).
|
|
uint32_t getBlockIdInKernel(int32_t Dim = DIM_X);
|
|
|
|
/// Return the number of blocks in the kernel.
|
|
uint32_t getNumberOfBlocksInKernel(int32_t Dim = DIM_X);
|
|
|
|
/// Return the kernel size, thus number of threads in the kernel.
|
|
uint32_t getNumberOfThreadsInKernel();
|
|
|
|
/// Return the maximal number of threads in the block usable for a team (=
|
|
/// parallel region).
|
|
///
|
|
/// Note: The version taking \p IsSPMD mode explicitly can be used during the
|
|
/// initialization of the target region, that is before `mapping::isSPMDMode()`
|
|
/// can be called by any thread other than the main one.
|
|
uint32_t getMaxTeamThreads();
|
|
uint32_t getMaxTeamThreads(bool IsSPMD);
|
|
|
|
/// Return the number of processing elements on the device.
|
|
uint32_t getNumberOfProcessorElements();
|
|
|
|
} // namespace mapping
|
|
|
|
} // namespace ompx
|
|
|
|
#endif
|