bob80905 created this revision.
bob80905 added reviewers: python3kgae, beanz.
Herald added a subscriber: Anastasia.
Herald added a project: All.
bob80905 requested review of this revision.
Herald added a project: clang.
Herald added a subscriber: cfe-commits.
This change exposes the pow library function for HLSL, only available for
floating point types.
The pow function is supported for all scalar, vector, and matrix types that
contain floating point types.
The full documentation of the HLSL pow function is available here:
https://learn.microsoft.com/en-us/windows/win32/direct3dhlsl/dx-graphics-hlsl-pow
Repository:
rG LLVM Github Monorepo
https://reviews.llvm.org/D156178
Files:
clang/lib/Headers/hlsl/hlsl_intrinsics.h
clang/test/CodeGenHLSL/builtins/pow.hlsl
Index: clang/test/CodeGenHLSL/builtins/pow.hlsl
===================================================================
--- /dev/null
+++ clang/test/CodeGenHLSL/builtins/pow.hlsl
@@ -0,0 +1,89 @@
+// RUN: %clang_cc1 -std=hlsl2021 -finclude-default-header -x hlsl -triple \
+// RUN: dxil-pc-shadermodel6.3-library %s -fnative-half-type \
+// RUN: -emit-llvm -disable-llvm-passes -O3 -o - | FileCheck %s
+// RUN: %clang_cc1 -std=hlsl2021 -finclude-default-header -x hlsl -triple \
+// RUN: dxil-pc-shadermodel6.3-library %s -emit-llvm -disable-llvm-passes \
+// RUN: -D__HLSL_ENABLE_16_BIT -o - | FileCheck %s --check-prefix=NO_HALF
+
+// CHECK: define noundef half @
+// CHECK: call half @llvm.pow.f16(
+// NO_HALF: define noundef float @"?test_pow_half@@YA$halff@$halff@0@Z"(
+// NO_HALF: call float @llvm.pow.f32(
+half test_pow_half(half p0, half p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <2 x half> @
+// CHECK: call <2 x half> @llvm.pow.v2f16
+// NO_HALF: define noundef <2 x float> @"?test_pow_float2@@YAT?$__vector@M$01@__clang@@T12@0@Z"(
+// NO_HALF: call <2 x float> @llvm.pow.v2f32(
+half2 test_pow_half2(half2 p0, half2 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <3 x half> @
+// CHECK: call <3 x half> @llvm.pow.v3f16
+// NO_HALF: define noundef <3 x float> @"?test_pow_float3@@YAT?$__vector@M$02@__clang@@T12@0@Z"(
+// NO_HALF: call <3 x float> @llvm.pow.v3f32(
+half3 test_pow_half3(half3 p0, half3 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <4 x half> @
+// CHECK: call <4 x half> @llvm.pow.v4f16
+// NO_HALF: define noundef <4 x float> @"?test_pow_float4@@YAT?$__vector@M$03@__clang@@T12@0@Z"(
+// NO_HALF: call <4 x float> @llvm.pow.v4f32(
+half4 test_pow_half4(half4 p0, half4 p1)
+{
+ return pow(p0, p1);
+}
+
+// CHECK: define noundef float @
+// CHECK: call float @llvm.pow.f32(
+float test_pow_float(float p0, float p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <2 x float> @
+// CHECK: call <2 x float> @llvm.pow.v2f32
+float2 test_pow_float2(float2 p0, float2 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <3 x float> @
+// CHECK: call <3 x float> @llvm.pow.v3f32
+float3 test_pow_float3(float3 p0, float3 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <4 x float> @
+// CHECK: call <4 x float> @llvm.pow.v4f32
+float4 test_pow_float4(float4 p0, float4 p1)
+{
+ return pow(p0, p1);
+}
+
+// CHECK: define noundef double @
+// CHECK: call double @llvm.pow.f64(
+double test_pow_double(double p0, double p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <2 x double> @
+// CHECK: call <2 x double> @llvm.pow.v2f64
+double2 test_pow_double2(double2 p0, double2 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <3 x double> @
+// CHECK: call <3 x double> @llvm.pow.v3f64
+double3 test_pow_double3(double3 p0, double3 p1)
+{
+ return pow(p0, p1);
+}
+// CHECK: define noundef <4 x double> @
+// CHECK: call <4 x double> @llvm.pow.v4f64
+double4 test_pow_double4(double4 p0, double4 p1)
+{
+ return pow(p0, p1);
+}
\ No newline at end of file
Index: clang/lib/Headers/hlsl/hlsl_intrinsics.h
===================================================================
--- clang/lib/Headers/hlsl/hlsl_intrinsics.h
+++ clang/lib/Headers/hlsl/hlsl_intrinsics.h
@@ -476,5 +476,89 @@
__attribute__((clang_builtin_alias(__builtin_elementwise_min)))
double4 min(double4, double4);
+// pow builtins
+#ifdef __HLSL_ENABLE_16_BIT
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+half pow(half, half);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+half2 pow(half2, half2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+half3 pow(half3, half3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+half4 pow(half4, half4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int16_t pow(int16_t, int16_t);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int16_t2 pow(int16_t2, int16_t2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int16_t3 pow(int16_t3, int16_t3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int16_t4 pow(int16_t4, int16_t4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint16_t pow(uint16_t, uint16_t);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint16_t2 pow(uint16_t2, uint16_t2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint16_t3 pow(uint16_t3, uint16_t3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint16_t4 pow(uint16_t4, uint16_t4);
+#endif
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow))) int pow(int,
+ int);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int2 pow(int2, int2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int3 pow(int3, int3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int4 pow(int4, int4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint pow(uint, uint);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint2 pow(uint2, uint2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint3 pow(uint3, uint3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint4 pow(uint4, uint4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int64_t pow(int64_t, int64_t);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int64_t2 pow(int64_t2, int64_t2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int64_t3 pow(int64_t3, int64_t3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+int64_t4 pow(int64_t4, int64_t4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint64_t pow(uint64_t, uint64_t);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint64_t2 pow(uint64_t2, uint64_t2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint64_t3 pow(uint64_t3, uint64_t3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+uint64_t4 pow(uint64_t4, uint64_t4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow))) float
+pow(float, float);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+float2 pow(float2, float2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+float3 pow(float3, float3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+float4 pow(float4, float4);
+
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow))) double
+pow(double, double);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+double2 pow(double2, double2);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+double3 pow(double3, double3);
+__attribute__((clang_builtin_alias(__builtin_elementwise_pow)))
+double4 pow(double4, double4);
+
} // namespace hlsl
#endif //_HLSL_HLSL_INTRINSICS_H_
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
