xref: /llvm-project/clang/test/CodeGenHLSL/builtins/sqrt.hlsl (revision 762f1b17b2815ccdfb4e5cb5412cb6210db92f73)
1// RUN: %clang_cc1 -finclude-default-header -triple dxil-pc-shadermodel6.3-library %s \
2// RUN:  -fnative-half-type -emit-llvm -disable-llvm-passes -o - | \
3// RUN:  FileCheck %s --check-prefixes=CHECK,NATIVE_HALF
4// RUN: %clang_cc1 -finclude-default-header -triple dxil-pc-shadermodel6.3-library %s \
5// RUN:  -emit-llvm -disable-llvm-passes -o - | \
6// RUN:  FileCheck %s --check-prefixes=CHECK,NO_HALF
7
8// NATIVE_HALF-LABEL: define noundef nofpclass(nan inf) half @_Z14test_sqrt_half
9// NATIVE_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn half @llvm.sqrt.f16(
10// NATIVE_HALF: ret half %{{.*}}
11// NO_HALF-LABEL: define noundef nofpclass(nan inf) float @_Z14test_sqrt_half
12// NO_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn float @llvm.sqrt.f32(
13// NO_HALF: ret float %{{.*}}
14half test_sqrt_half(half p0) { return sqrt(p0); }
15// NATIVE_HALF-LABEL: define noundef nofpclass(nan inf) <2 x half> @_Z15test_sqrt_half2
16// NATIVE_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <2 x half> @llvm.sqrt.v2f16
17// NATIVE_HALF: ret <2 x half> %{{.*}}
18// NO_HALF-LABEL: define noundef nofpclass(nan inf) <2 x float> @_Z15test_sqrt_half2
19// NO_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <2 x float> @llvm.sqrt.v2f32(
20// NO_HALF: ret <2 x float> %{{.*}}
21half2 test_sqrt_half2(half2 p0) { return sqrt(p0); }
22// NATIVE_HALF-LABEL: define noundef nofpclass(nan inf) <3 x half> @_Z15test_sqrt_half3
23// NATIVE_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <3 x half> @llvm.sqrt.v3f16
24// NATIVE_HALF: ret <3 x half> %{{.*}}
25// NO_HALF-LABEL: define noundef nofpclass(nan inf) <3 x float> @_Z15test_sqrt_half3
26// NO_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <3 x float> @llvm.sqrt.v3f32(
27// NO_HALF: ret <3 x float> %{{.*}}
28half3 test_sqrt_half3(half3 p0) { return sqrt(p0); }
29// NATIVE_HALF-LABEL: define noundef nofpclass(nan inf) <4 x half> @_Z15test_sqrt_half4
30// NATIVE_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <4 x half> @llvm.sqrt.v4f16
31// NATIVE_HALF: ret <4 x half> %{{.*}}
32// NO_HALF-LABEL: define noundef nofpclass(nan inf) <4 x float> @_Z15test_sqrt_half4
33// NO_HALF: %{{.*}} = call reassoc nnan ninf nsz arcp afn <4 x float> @llvm.sqrt.v4f32(
34// NO_HALF: ret <4 x float> %{{.*}}
35half4 test_sqrt_half4(half4 p0) { return sqrt(p0); }
36
37// CHECK-LABEL: define noundef nofpclass(nan inf) float @_Z15test_sqrt_float
38// CHECK: %{{.*}} = call reassoc nnan ninf nsz arcp afn float @llvm.sqrt.f32(
39// CHECK: ret float %{{.*}}
40float test_sqrt_float(float p0) { return sqrt(p0); }
41// CHECK-LABEL: define noundef nofpclass(nan inf) <2 x float> @_Z16test_sqrt_float2
42// CHECK: %{{.*}} = call reassoc nnan ninf nsz arcp afn <2 x float> @llvm.sqrt.v2f32
43// CHECK: ret <2 x float> %{{.*}}
44float2 test_sqrt_float2(float2 p0) { return sqrt(p0); }
45// CHECK-LABEL: define noundef nofpclass(nan inf) <3 x float> @_Z16test_sqrt_float3
46// CHECK: %{{.*}} = call reassoc nnan ninf nsz arcp afn <3 x float> @llvm.sqrt.v3f32
47// CHECK: ret <3 x float> %{{.*}}
48float3 test_sqrt_float3(float3 p0) { return sqrt(p0); }
49// CHECK-LABEL: define noundef nofpclass(nan inf) <4 x float> @_Z16test_sqrt_float4
50// CHECK: %{{.*}} = call reassoc nnan ninf nsz arcp afn <4 x float> @llvm.sqrt.v4f32
51// CHECK: ret <4 x float> %{{.*}}
52float4 test_sqrt_float4(float4 p0) { return sqrt(p0); }
53