160ff83765
Provides AMDGCN and NVPTX specific specialization of getGPUWarpSize, getGPUThreadID, and getGPUNumThreads methods. Adds tests for AMDGCN codegen for these methods in generic and simd modes. Also changes the precondition in InitTempAlloca to be slightly more permissive. Useful for AMDGCN OpenMP codegen where allocas are created with a cast to an address space. Reviewed By: ABataev Differential Revision: https://reviews.llvm.org/D84260
44 lines
1.4 KiB
C++
44 lines
1.4 KiB
C++
//===--- CGOpenMPRuntimeAMDGCN.h - Interface to OpenMP AMDGCN Runtimes ---===//
|
|
//
|
|
// 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 provides a class for OpenMP runtime code generation specialized to
|
|
// AMDGCN targets from generalized CGOpenMPRuntimeGPU class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_LIB_CODEGEN_CGOPENMPRUNTIMEAMDGCN_H
|
|
#define LLVM_CLANG_LIB_CODEGEN_CGOPENMPRUNTIMEAMDGCN_H
|
|
|
|
#include "CGOpenMPRuntime.h"
|
|
#include "CGOpenMPRuntimeGPU.h"
|
|
#include "CodeGenFunction.h"
|
|
#include "clang/AST/StmtOpenMP.h"
|
|
|
|
namespace clang {
|
|
namespace CodeGen {
|
|
|
|
class CGOpenMPRuntimeAMDGCN final : public CGOpenMPRuntimeGPU {
|
|
|
|
public:
|
|
explicit CGOpenMPRuntimeAMDGCN(CodeGenModule &CGM);
|
|
|
|
/// Get the GPU warp size.
|
|
llvm::Value *getGPUWarpSize(CodeGenFunction &CGF) override;
|
|
|
|
/// Get the id of the current thread on the GPU.
|
|
llvm::Value *getGPUThreadID(CodeGenFunction &CGF) override;
|
|
|
|
/// Get the maximum number of threads in a block of the GPU.
|
|
llvm::Value *getGPUNumThreads(CodeGenFunction &CGF) override;
|
|
};
|
|
|
|
} // namespace CodeGen
|
|
} // namespace clang
|
|
|
|
#endif // LLVM_CLANG_LIB_CODEGEN_CGOPENMPRUNTIMEAMDGCN_H
|