maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Simplify the implementation of getCUDALibDeviceFunction. NFC. 

Summary:
The function is currently awfully complicated. Drop the IILE and use
StringRef over std::string.

Reviewers: Meinersbur, grosser, bollu

Reviewed By: Meinersbur

Subscribers: nemanjai, kbarton, bollu, llvm-commits, pollydev

Differential Revision: https://reviews.llvm.org/D48070

llvm-svn: 334695
This commit is contained in:
Philip Pfaffe 2018-06-14 08:54:55 +00:00
parent 7d1448ea72
commit 66a05ad672
1 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
polly/lib/CodeGen/PPCGCodeGeneration.cpp
View File

@ -1398,7 +1398,7 @@ const std::map<std::string, std::string> IntrinsicToLibdeviceFunc = {
{"llvm.powi.f64", "powi"},
{"llvm.powi.f32", "powif"}};
/// Return the corresponding CUDA libdevice function name for @p F.
/// Return the corresponding CUDA libdevice function name @p Name.
/// Note that this function will try to convert instrinsics in the list
/// IntrinsicToLibdeviceFunc into libdevice functions.
/// This is because some intrinsics such as `exp`
@ -1407,17 +1407,13 @@ const std::map<std::string, std::string> IntrinsicToLibdeviceFunc = {
/// so that we use intrinsics whenever possible.
///
/// Return "" if we are not compiling for CUDA.
std::string getCUDALibDeviceFuntion(Function *F) {
auto FnName = [&]() -> const std::string {
auto It = IntrinsicToLibdeviceFunc.find(F->getName());
if (It != IntrinsicToLibdeviceFunc.end())
return It->second;
std::string getCUDALibDeviceFuntion(StringRef Name) {
auto It = IntrinsicToLibdeviceFunc.find(Name);
if (It != IntrinsicToLibdeviceFunc.end())
return getCUDALibDeviceFuntion(It->second);
return std::string(F->getName());
}();
if (CUDALibDeviceFunctions.count(FnName))
return "__nv_" + FnName;
if (CUDALibDeviceFunctions.count(Name))
return ("__nv_" + Name).str();
return "";
}
@ -1430,7 +1426,7 @@ static bool isValidFunctionInKernel(llvm::Function *F, bool AllowLibDevice) {
// "llvm.copysign".
const StringRef Name = F->getName();
if (AllowLibDevice && getCUDALibDeviceFuntion(F).length() > 0)
if (AllowLibDevice && getCUDALibDeviceFuntion(Name).length() > 0)
return true;
return F->isIntrinsic() &&
@ -2389,7 +2385,7 @@ bool GPUNodeBuilder::requiresCUDALibDevice() {
if (!F.isDeclaration())
continue;
const std::string CUDALibDeviceFunc = getCUDALibDeviceFuntion(&F);
const std::string CUDALibDeviceFunc = getCUDALibDeviceFuntion(F.getName());
if (CUDALibDeviceFunc.length() != 0) {
// We need to handle the case where a module looks like this:
// @expf(..)