[clang] [AMDGPU][clang] provide device implementation for __builtin_logb and … (PR #129347)
via cfe-commits
cfe-commits at lists.llvm.org
Fri Mar 14 10:44:33 PDT 2025
https://github.com/choikwa updated https://github.com/llvm/llvm-project/pull/129347
>From 2b45fac1d2c21f064fa8bb8690fc70a44fcfc851 Mon Sep 17 00:00:00 2001
From: Kevin Choi <kevin.choi at amd.com>
Date: Fri, 28 Feb 2025 16:52:03 -0600
Subject: [PATCH] [AMDGPU][clang] provide device implementation for
__builtin_logb and __builtin_scalbn
Clang generates library calls for __builtin_* functions which can be a problem for GPUs that cannot handle them.
This patch generates a device implementations for __builtin_logb and __builtin_scalbn.
---
clang/lib/CodeGen/CGBuiltin.cpp | 43 +++++++++++++++++-
clang/lib/CodeGen/CodeGenModule.h | 5 +++
clang/test/CodeGenHIP/logb_scalbn.hip | 65 +++++++++++++++++++++++++++
3 files changed, 112 insertions(+), 1 deletion(-)
create mode 100644 clang/test/CodeGenHIP/logb_scalbn.hip
diff --git a/clang/lib/CodeGen/CGBuiltin.cpp b/clang/lib/CodeGen/CGBuiltin.cpp
index a5ed2595bad4d..abc40d15bfa25 100644
--- a/clang/lib/CodeGen/CGBuiltin.cpp
+++ b/clang/lib/CodeGen/CGBuiltin.cpp
@@ -6621,10 +6621,27 @@ RValue CodeGenFunction::EmitBuiltinExpr(const GlobalDecl GD, unsigned BuiltinID,
}
}
+ // These will be emitted as Intrinsic later.
+ auto NeedsDeviceOverloadToIntrin = [&](unsigned BuiltinID) {
+ if (getTarget().getTriple().isAMDGCN()) {
+ switch (BuiltinID) {
+ default:
+ return false;
+ case Builtin::BIlogb:
+ case Builtin::BI__builtin_logb:
+ case Builtin::BIscalbn:
+ case Builtin::BI__builtin_scalbn:
+ return true;
+ }
+ }
+ return false;
+ };
+
// If this is an alias for a lib function (e.g. __builtin_sin), emit
// the call using the normal call path, but using the unmangled
// version of the function name.
- if (getContext().BuiltinInfo.isLibFunction(BuiltinID))
+ if (!NeedsDeviceOverloadToIntrin(BuiltinID) &&
+ getContext().BuiltinInfo.isLibFunction(BuiltinID))
return emitLibraryCall(*this, FD, E,
CGM.getBuiltinLibFunction(FD, BuiltinID));
@@ -20910,6 +20927,30 @@ Value *CodeGenFunction::EmitAMDGPUBuiltinExpr(unsigned BuiltinID,
case AMDGPU::BI__builtin_amdgcn_s_prefetch_data:
return emitBuiltinWithOneOverloadedType<2>(
*this, E, Intrinsic::amdgcn_s_prefetch_data);
+ case Builtin::BIlogb:
+ case Builtin::BI__builtin_logb: {
+ auto Src0 = EmitScalarExpr(E->getArg(0));
+ auto FrExpFunc = CGM.getIntrinsic(Intrinsic::amdgcn_frexp_exp,
+ { Builder.getInt32Ty(), Src0->getType() });
+ auto FrExp = Builder.CreateCall(FrExpFunc, Src0);
+ auto Add = Builder.CreateAdd(
+ FrExp, ConstantInt::getSigned(FrExp->getType(), -1), "", false, true);
+ auto SIToFP = Builder.CreateSIToFP(Add, Builder.getDoubleTy());
+ auto Fabs = emitBuiltinWithOneOverloadedType<1>(*this, E, Intrinsic::fabs);
+ auto FCmpONE = Builder.CreateFCmpONE(
+ Fabs, ConstantFP::getInfinity(Builder.getDoubleTy()));
+ auto Sel1 = Builder.CreateSelect(FCmpONE, SIToFP, Fabs);
+ auto FCmpOEQ = Builder.CreateFCmpOEQ(
+ Src0, ConstantFP::getZero(Builder.getDoubleTy()));
+ auto Sel2 = Builder.CreateSelect(
+ FCmpOEQ, ConstantFP::getInfinity(Builder.getDoubleTy(), /*Neg*/true),
+ Sel1);
+ return Sel2;
+ }
+ case Builtin::BIscalbn:
+ case Builtin::BI__builtin_scalbn:
+ return emitBinaryExpMaybeConstrainedFPBuiltin(
+ *this, E, Intrinsic::ldexp, Intrinsic::experimental_constrained_ldexp);
default:
return nullptr;
}
diff --git a/clang/lib/CodeGen/CodeGenModule.h b/clang/lib/CodeGen/CodeGenModule.h
index 6deb467b2cc9f..ae1441dc3d009 100644
--- a/clang/lib/CodeGen/CodeGenModule.h
+++ b/clang/lib/CodeGen/CodeGenModule.h
@@ -1231,6 +1231,11 @@ class CodeGenModule : public CodeGenTypeCache {
llvm::FunctionType *FnType = nullptr, bool DontDefer = false,
ForDefinition_t IsForDefinition = NotForDefinition);
+ /// Given a builtin id for a function, return a Function* for device
+ /// overload implementation.
+ llvm::Constant *getDeviceLibFunction(const FunctionDecl *FD,
+ unsigned BuiltinID);
+
/// Given a builtin id for a function like "__builtin_fabsf", return a
/// Function* for "fabsf".
llvm::Constant *getBuiltinLibFunction(const FunctionDecl *FD,
diff --git a/clang/test/CodeGenHIP/logb_scalbn.hip b/clang/test/CodeGenHIP/logb_scalbn.hip
new file mode 100644
index 0000000000000..9d284a36c1de0
--- /dev/null
+++ b/clang/test/CodeGenHIP/logb_scalbn.hip
@@ -0,0 +1,65 @@
+// NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py UTC_ARGS: --version 5
+// RUN: %clang --cuda-device-only -nogpulib -emit-llvm -S -o - %s | FileCheck %s
+#include <math.h>
+#define __device__ __attribute__((device))
+
+// CHECK-LABEL: define hidden void @_Z9my_kernelv(
+// CHECK-SAME: ) #[[ATTR2:[0-9]+]] {
+// CHECK-NEXT: [[ENTRY:.*:]]
+// CHECK-NEXT: [[RETVAL_I8:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[__X_ADDR_I9:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[RETVAL_I4:%.*]] = alloca double, align 8, addrspace(5)
+// CHECK-NEXT: [[__X_ADDR_I5:%.*]] = alloca double, align 8, addrspace(5)
+// CHECK-NEXT: [[__N_ADDR_I:%.*]] = alloca i32, align 4, addrspace(5)
+// CHECK-NEXT: [[RETVAL_I:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[__X_ADDR_I:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[D1:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[D2:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[D3:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[D4:%.*]] = alloca float, align 4, addrspace(5)
+// CHECK-NEXT: [[D1_ASCAST:%.*]] = addrspacecast ptr addrspace(5) [[D1]] to ptr
+// CHECK-NEXT: [[D2_ASCAST:%.*]] = addrspacecast ptr addrspace(5) [[D2]] to ptr
+// CHECK-NEXT: [[D3_ASCAST:%.*]] = addrspacecast ptr addrspace(5) [[D3]] to ptr
+// CHECK-NEXT: [[D4_ASCAST:%.*]] = addrspacecast ptr addrspace(5) [[D4]] to ptr
+// CHECK-NEXT: [[TMP0:%.*]] = call i32 @llvm.amdgcn.frexp.exp.i32.f64(double 1.600000e+01)
+// CHECK-NEXT: [[TMP1:%.*]] = add nsw i32 [[TMP0]], -1
+// CHECK-NEXT: [[TMP2:%.*]] = sitofp i32 [[TMP1]] to double
+// CHECK-NEXT: [[TMP3:%.*]] = call contract double @llvm.fabs.f64(double 1.600000e+01)
+// CHECK-NEXT: [[TMP4:%.*]] = fcmp contract one double [[TMP3]], 0x7FF0000000000000
+// CHECK-NEXT: [[TMP5:%.*]] = select contract i1 [[TMP4]], double [[TMP2]], double [[TMP3]]
+// CHECK-NEXT: [[TMP6:%.*]] = select contract i1 false, double 0xFFF0000000000000, double [[TMP5]]
+// CHECK-NEXT: [[CONV:%.*]] = fptrunc contract double [[TMP6]] to float
+// CHECK-NEXT: store float [[CONV]], ptr [[D1_ASCAST]], align 4
+// CHECK-NEXT: [[RETVAL_ASCAST_I:%.*]] = addrspacecast ptr addrspace(5) [[RETVAL_I]] to ptr
+// CHECK-NEXT: [[__X_ADDR_ASCAST_I:%.*]] = addrspacecast ptr addrspace(5) [[__X_ADDR_I]] to ptr
+// CHECK-NEXT: store float 1.600000e+01, ptr [[__X_ADDR_ASCAST_I]], align 4
+// CHECK-NEXT: [[TMP7:%.*]] = load float, ptr [[__X_ADDR_ASCAST_I]], align 4
+// CHECK-NEXT: [[RETVAL_ASCAST_I10:%.*]] = addrspacecast ptr addrspace(5) [[RETVAL_I8]] to ptr
+// CHECK-NEXT: [[__X_ADDR_ASCAST_I11:%.*]] = addrspacecast ptr addrspace(5) [[__X_ADDR_I9]] to ptr
+// CHECK-NEXT: store float [[TMP7]], ptr [[__X_ADDR_ASCAST_I11]], align 4
+// CHECK-NEXT: [[TMP8:%.*]] = load float, ptr [[__X_ADDR_ASCAST_I11]], align 4
+// CHECK-NEXT: [[CALL_I12:%.*]] = call contract noundef float @__ocml_logb_f32(float noundef [[TMP8]]) #[[ATTR5:[0-9]+]]
+// CHECK-NEXT: store float [[CALL_I12]], ptr [[D2_ASCAST]], align 4
+// CHECK-NEXT: [[TMP9:%.*]] = call contract double @llvm.ldexp.f64.i32(double 1.600000e+01, i32 10)
+// CHECK-NEXT: [[CONV1:%.*]] = fptrunc contract double [[TMP9]] to float
+// CHECK-NEXT: store float [[CONV1]], ptr [[D3_ASCAST]], align 4
+// CHECK-NEXT: [[RETVAL_ASCAST_I6:%.*]] = addrspacecast ptr addrspace(5) [[RETVAL_I4]] to ptr
+// CHECK-NEXT: [[__X_ADDR_ASCAST_I7:%.*]] = addrspacecast ptr addrspace(5) [[__X_ADDR_I5]] to ptr
+// CHECK-NEXT: [[__N_ADDR_ASCAST_I:%.*]] = addrspacecast ptr addrspace(5) [[__N_ADDR_I]] to ptr
+// CHECK-NEXT: store double 9.000000e+00, ptr [[__X_ADDR_ASCAST_I7]], align 8
+// CHECK-NEXT: store i32 4, ptr [[__N_ADDR_ASCAST_I]], align 4
+// CHECK-NEXT: [[TMP10:%.*]] = load double, ptr [[__X_ADDR_ASCAST_I7]], align 8
+// CHECK-NEXT: [[TMP11:%.*]] = load i32, ptr [[__N_ADDR_ASCAST_I]], align 4
+// CHECK-NEXT: [[TMP12:%.*]] = call contract noundef double @llvm.ldexp.f64.i32(double [[TMP10]], i32 [[TMP11]])
+// CHECK-NEXT: [[CONV3:%.*]] = fptrunc contract double [[TMP12]] to float
+// CHECK-NEXT: store float [[CONV3]], ptr [[D4_ASCAST]], align 4
+// CHECK-NEXT: ret void
+//
+__device__ void my_kernel(){
+
+ float D1 = __builtin_logb((float)16);
+ float D2 = logb((float)16);
+ float D3 = __builtin_scalbn((float)16, 10);
+ float D4 = scalbn(9.0, 4.0);
+}
+
More information about the cfe-commits
mailing list