xref: /llvm-project/clang/test/CodeGenHLSL/builtins/rsqrt.hlsl (revision 762f1b17b2815ccdfb4e5cb5412cb6210db92f73)
1// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple \
2// RUN:   dxil-pc-shadermodel6.3-library %s -fnative-half-type \
3// RUN:   -emit-llvm -disable-llvm-passes -o - | FileCheck %s \
4// RUN:   --check-prefixes=CHECK,NATIVE_HALF \
5// RUN:   -DFNATTRS="noundef nofpclass(nan inf)" -DTARGET=dx
6// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple \
7// RUN:   dxil-pc-shadermodel6.3-library %s -emit-llvm -disable-llvm-passes \
8// RUN:   -o - | FileCheck %s --check-prefixes=CHECK,NO_HALF \
9// RUN:   -DFNATTRS="noundef nofpclass(nan inf)" -DTARGET=dx
10// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple \
11// RUN:   spirv-unknown-vulkan-compute %s -fnative-half-type \
12// RUN:   -emit-llvm -disable-llvm-passes -o - | FileCheck %s \
13// RUN:   --check-prefixes=CHECK,NATIVE_HALF \
14// RUN:   -DFNATTRS="spir_func noundef nofpclass(nan inf)" -DTARGET=spv
15// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple \
16// RUN:   spirv-unknown-vulkan-compute %s -emit-llvm -disable-llvm-passes \
17// RUN:   -o - | FileCheck %s --check-prefixes=CHECK,NO_HALF \
18// RUN:   -DFNATTRS="spir_func noundef nofpclass(nan inf)" -DTARGET=spv
19
20// NATIVE_HALF: define [[FNATTRS]] half @
21// NATIVE_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn half @llvm.[[TARGET]].rsqrt.f16(
22// NATIVE_HALF: ret half %hlsl.rsqrt
23// NO_HALF: define [[FNATTRS]] float @
24// NO_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn float @llvm.[[TARGET]].rsqrt.f32(
25// NO_HALF: ret float %hlsl.rsqrt
26half test_rsqrt_half(half p0) { return rsqrt(p0); }
27// NATIVE_HALF: define [[FNATTRS]] <2 x half> @
28// NATIVE_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <2 x half> @llvm.[[TARGET]].rsqrt.v2f16
29// NATIVE_HALF: ret <2 x half> %hlsl.rsqrt
30// NO_HALF: define [[FNATTRS]] <2 x float> @
31// NO_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <2 x float> @llvm.[[TARGET]].rsqrt.v2f32(
32// NO_HALF: ret <2 x float> %hlsl.rsqrt
33half2 test_rsqrt_half2(half2 p0) { return rsqrt(p0); }
34// NATIVE_HALF: define [[FNATTRS]] <3 x half> @
35// NATIVE_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <3 x half> @llvm.[[TARGET]].rsqrt.v3f16
36// NATIVE_HALF: ret <3 x half> %hlsl.rsqrt
37// NO_HALF: define [[FNATTRS]] <3 x float> @
38// NO_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <3 x float> @llvm.[[TARGET]].rsqrt.v3f32(
39// NO_HALF: ret <3 x float> %hlsl.rsqrt
40half3 test_rsqrt_half3(half3 p0) { return rsqrt(p0); }
41// NATIVE_HALF: define [[FNATTRS]] <4 x half> @
42// NATIVE_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <4 x half> @llvm.[[TARGET]].rsqrt.v4f16
43// NATIVE_HALF: ret <4 x half> %hlsl.rsqrt
44// NO_HALF: define [[FNATTRS]] <4 x float> @
45// NO_HALF: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <4 x float> @llvm.[[TARGET]].rsqrt.v4f32(
46// NO_HALF: ret <4 x float> %hlsl.rsqrt
47half4 test_rsqrt_half4(half4 p0) { return rsqrt(p0); }
48
49// CHECK: define [[FNATTRS]] float @
50// CHECK: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn float @llvm.[[TARGET]].rsqrt.f32(
51// CHECK: ret float %hlsl.rsqrt
52float test_rsqrt_float(float p0) { return rsqrt(p0); }
53// CHECK: define [[FNATTRS]] <2 x float> @
54// CHECK: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <2 x float> @llvm.[[TARGET]].rsqrt.v2f32
55// CHECK: ret <2 x float> %hlsl.rsqrt
56float2 test_rsqrt_float2(float2 p0) { return rsqrt(p0); }
57// CHECK: define [[FNATTRS]] <3 x float> @
58// CHECK: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <3 x float> @llvm.[[TARGET]].rsqrt.v3f32
59// CHECK: ret <3 x float> %hlsl.rsqrt
60float3 test_rsqrt_float3(float3 p0) { return rsqrt(p0); }
61// CHECK: define [[FNATTRS]] <4 x float> @
62// CHECK: %hlsl.rsqrt = call reassoc nnan ninf nsz arcp afn <4 x float> @llvm.[[TARGET]].rsqrt.v4f32
63// CHECK: ret <4 x float> %hlsl.rsqrt
64float4 test_rsqrt_float4(float4 p0) { return rsqrt(p0); }
65