b6be44ad0d
In SPIR-V, kernel arguments are not allowed to be in the Generic AS, in both Intel's internal SPIR-V offloading implementation as well as HIPSPV, `CrossWorkgroup` AS1 is used. Do the same for OMPSPV. Currently with Generic AS the `llvm-spirv` translator blows up if we are using it, and if not, the GPU runtime blows up. To get the existing logic to set the correct AS to kick in, we need to know if the function is a kernel or not at the time we first create the function that may end up as the kernel. I use the existing `arrangeSYCLKernelCallerDeclaration` function to do the right kernel ABI computation, but since the function is not specific to SYCL anymore because I merged all the device kernel clang attributes into one. Rename the function to be accurate to the current behavior, `arrangeDeviceKernelCallerDeclaration`. --------- Signed-off-by: Sarnie, Nick <nick.sarnie@intel.com>
73 lines
3.0 KiB
C++
73 lines
3.0 KiB
C++
//===--------- CodeGenSYCL.cpp - Code for SYCL kernel generation ----------===//
|
|
//
|
|
// 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 contains code required for generation of SYCL kernel caller offload
|
|
// entry point functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "CodeGenFunction.h"
|
|
#include "CodeGenModule.h"
|
|
|
|
using namespace clang;
|
|
using namespace CodeGen;
|
|
|
|
static void SetSYCLKernelAttributes(llvm::Function *Fn, CodeGenFunction &CGF) {
|
|
// SYCL 2020 device language restrictions require forward progress and
|
|
// disallow recursion.
|
|
Fn->setDoesNotRecurse();
|
|
if (CGF.checkIfFunctionMustProgress())
|
|
Fn->addFnAttr(llvm::Attribute::MustProgress);
|
|
}
|
|
|
|
void CodeGenModule::EmitSYCLKernelCaller(const FunctionDecl *KernelEntryPointFn,
|
|
ASTContext &Ctx) {
|
|
assert(Ctx.getLangOpts().SYCLIsDevice &&
|
|
"SYCL kernel caller offload entry point functions can only be emitted"
|
|
" during device compilation");
|
|
|
|
const auto *KernelEntryPointAttr =
|
|
KernelEntryPointFn->getAttr<SYCLKernelEntryPointAttr>();
|
|
assert(KernelEntryPointAttr && "Missing sycl_kernel_entry_point attribute");
|
|
assert(!KernelEntryPointAttr->isInvalidAttr() &&
|
|
"sycl_kernel_entry_point attribute is invalid");
|
|
|
|
// Find the SYCLKernelCallStmt.
|
|
SYCLKernelCallStmt *KernelCallStmt =
|
|
cast<SYCLKernelCallStmt>(KernelEntryPointFn->getBody());
|
|
|
|
// Retrieve the SYCL kernel caller parameters from the OutlinedFunctionDecl.
|
|
FunctionArgList Args;
|
|
const OutlinedFunctionDecl *OutlinedFnDecl =
|
|
KernelCallStmt->getOutlinedFunctionDecl();
|
|
Args.append(OutlinedFnDecl->param_begin(), OutlinedFnDecl->param_end());
|
|
|
|
// Compute the function info and LLVM function type.
|
|
const CGFunctionInfo &FnInfo =
|
|
getTypes().arrangeDeviceKernelCallerDeclaration(Ctx.VoidTy, Args);
|
|
llvm::FunctionType *FnTy = getTypes().GetFunctionType(FnInfo);
|
|
|
|
// Retrieve the generated name for the SYCL kernel caller function.
|
|
CanQualType KernelNameType =
|
|
Ctx.getCanonicalType(KernelEntryPointAttr->getKernelName());
|
|
const SYCLKernelInfo &KernelInfo = Ctx.getSYCLKernelInfo(KernelNameType);
|
|
auto *Fn = llvm::Function::Create(FnTy, llvm::Function::ExternalLinkage,
|
|
KernelInfo.GetKernelName(), &getModule());
|
|
|
|
// Emit the SYCL kernel caller function.
|
|
CodeGenFunction CGF(*this);
|
|
SetLLVMFunctionAttributes(GlobalDecl(), FnInfo, Fn, false);
|
|
SetSYCLKernelAttributes(Fn, CGF);
|
|
CGF.StartFunction(GlobalDecl(), Ctx.VoidTy, Fn, FnInfo, Args,
|
|
SourceLocation(), SourceLocation());
|
|
CGF.EmitFunctionBody(OutlinedFnDecl->getBody());
|
|
setDSOLocal(Fn);
|
|
SetLLVMFunctionAttributesForDefinition(cast<Decl>(OutlinedFnDecl), Fn);
|
|
CGF.FinishFunction();
|
|
}
|