xref: /llvm-project/clang/test/CodeGenHLSL/builtins/splitdouble.hlsl (revision 481bce018ea8872277f79102842eaf8a55f634a2)
1// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple dxil-pc-shadermodel6.3-library %s -fnative-half-type -emit-llvm -O1 -o - | FileCheck %s
2// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple spirv-vulkan-library %s -fnative-half-type -emit-llvm -O0 -o - | FileCheck %s --check-prefix=SPIRV
3
4
5
6// CHECK: define {{.*}} i32 {{.*}}test_scalar{{.*}}(double {{.*}} [[VALD:%.*]])
7// CHECK:      [[VALRET:%.*]] = {{.*}} call { i32, i32 } @llvm.dx.splitdouble.i32(double [[VALD]])
8// CHECK-NEXT: extractvalue { i32, i32 } [[VALRET]], 0
9// CHECK-NEXT: extractvalue { i32, i32 } [[VALRET]], 1
10//
11// SPIRV: define spir_func {{.*}} i32 {{.*}}test_scalar{{.*}}(double {{.*}} [[VALD:%.*]])
12// SPIRV-NOT:  @llvm.dx.splitdouble.i32
13// SPIRV:      [[LOAD:%.*]] = load double, ptr [[VALD]].addr, align 8
14// SPIRV-NEXT: [[CAST:%.*]] = bitcast double [[LOAD]] to <2 x i32>
15// SPIRV-NEXT: extractelement <2 x i32> [[CAST]], i64 0
16// SPIRV-NEXT: extractelement <2 x i32> [[CAST]], i64 1
17uint test_scalar(double D) {
18  uint A, B;
19  asuint(D, A, B);
20  return A + B;
21}
22
23// CHECK: define {{.*}} <1 x i32> {{.*}}test_double1{{.*}}(<1 x double> {{.*}} [[VALD:%.*]])
24// CHECK:      [[TRUNC:%.*]] = extractelement <1 x double> %D, i64 0
25// CHECK-NEXT: [[VALRET:%.*]] = {{.*}} call { i32, i32 } @llvm.dx.splitdouble.i32(double [[TRUNC]])
26// CHECK-NEXT: extractvalue { i32, i32 } [[VALRET]], 0
27// CHECK-NEXT: extractvalue { i32, i32 } [[VALRET]], 1
28//
29// SPIRV: define spir_func {{.*}} <1 x i32> {{.*}}test_double1{{.*}}(<1 x double> {{.*}} [[VALD:%.*]])
30// SPIRV-NOT:  @llvm.dx.splitdouble.i32
31// SPIRV:      [[LOAD:%.*]] = load <1 x double>, ptr [[VALD]].addr, align 8
32// SPIRV-NEXT: [[TRUNC:%.*]] = extractelement <1 x double> [[LOAD]], i64 0
33// SPIRV-NEXT: [[CAST:%.*]] = bitcast double [[TRUNC]] to <2 x i32>
34// SPIRV-NEXT: extractelement <2 x i32> [[CAST]], i64 0
35// SPIRV-NEXT: extractelement <2 x i32> [[CAST]], i64 1
36uint1 test_double1(double1 D) {
37  uint A, B;
38  asuint(D, A, B);
39  return A + B;
40}
41
42// CHECK: define {{.*}} <2 x i32> {{.*}}test_vector2{{.*}}(<2 x double> {{.*}} [[VALD:%.*]])
43// CHECK:      [[VALRET:%.*]] = {{.*}} call { <2 x i32>, <2 x i32> } @llvm.dx.splitdouble.v2i32(<2 x double> [[VALD]])
44// CHECK-NEXT: extractvalue { <2 x i32>, <2 x i32> } [[VALRET]], 0
45// CHECK-NEXT: extractvalue { <2 x i32>, <2 x i32> } [[VALRET]], 1
46//
47// SPIRV: define spir_func {{.*}} <2 x i32> {{.*}}test_vector2{{.*}}(<2 x double> {{.*}} [[VALD:%.*]])
48// SPIRV-NOT:  @llvm.dx.splitdouble.i32
49// SPIRV:      [[LOAD:%.*]] = load <2 x double>, ptr [[VALD]].addr, align 16
50// SPIRV-NEXT: [[CAST1:%.*]] = bitcast <2 x double> [[LOAD]] to <4 x i32>
51// SPIRV-NEXT: [[SHUF1:%.*]] = shufflevector <4 x i32> [[CAST1]], <4 x i32> poison, <2 x i32> <i32 0, i32 2>
52// SPIRV-NEXT: [[SHUF2:%.*]] = shufflevector <4 x i32> [[CAST1]], <4 x i32> poison, <2 x i32> <i32 1, i32 3>
53uint2 test_vector2(double2 D) {
54  uint2 A, B;
55  asuint(D, A, B);
56  return A + B;
57}
58
59// CHECK: define {{.*}} <3 x i32> {{.*}}test_vector3{{.*}}(<3 x double> {{.*}} [[VALD:%.*]])
60// CHECK:      [[VALRET:%.*]] = {{.*}} call { <3 x i32>, <3 x i32> } @llvm.dx.splitdouble.v3i32(<3 x double> [[VALD]])
61// CHECK-NEXT: extractvalue { <3 x i32>, <3 x i32> } [[VALRET]], 0
62// CHECK-NEXT: extractvalue { <3 x i32>, <3 x i32> } [[VALRET]], 1
63//
64// SPIRV: define spir_func {{.*}} <3 x i32> {{.*}}test_vector3{{.*}}(<3 x double> {{.*}} [[VALD:%.*]])
65// SPIRV-NOT:  @llvm.dx.splitdouble.i32
66// SPIRV:      [[LOAD:%.*]] = load <3 x double>, ptr [[VALD]].addr, align 32
67// SPIRV-NEXT: [[CAST1:%.*]] = bitcast <3 x double> [[LOAD]] to <6 x i32>
68// SPIRV-NEXT: [[SHUF1:%.*]] = shufflevector <6 x i32> [[CAST1]], <6 x i32> poison, <3 x i32> <i32 0, i32 2, i32 4>
69// SPIRV-NEXT: [[SHUF2:%.*]] = shufflevector <6 x i32> [[CAST1]], <6 x i32> poison, <3 x i32> <i32 1, i32 3, i32 5>
70uint3 test_vector3(double3 D) {
71  uint3 A, B;
72  asuint(D, A, B);
73  return A + B;
74}
75
76// CHECK: define {{.*}} <4 x i32> {{.*}}test_vector4{{.*}}(<4 x double> {{.*}} [[VALD:%.*]])
77// CHECK:      [[VALRET:%.*]] = {{.*}} call { <4 x i32>, <4 x i32> } @llvm.dx.splitdouble.v4i32(<4 x double> [[VALD]])
78// CHECK-NEXT: extractvalue { <4 x i32>, <4 x i32> } [[VALRET]], 0
79// CHECK-NEXT: extractvalue { <4 x i32>, <4 x i32> } [[VALRET]], 1
80//
81// SPIRV: define spir_func {{.*}} <4 x i32> {{.*}}test_vector4{{.*}}(<4 x double> {{.*}} [[VALD:%.*]])
82// SPIRV-NOT: @llvm.dx.splitdouble.i32
83// SPIRV:      [[LOAD:%.*]] = load <4 x double>, ptr [[VALD]].addr, align 32
84// SPIRV-NEXT: [[CAST1:%.*]] = bitcast <4 x double> [[LOAD]] to <8 x i32>
85// SPIRV-NEXT: [[SHUF1:%.*]] = shufflevector <8 x i32> [[CAST1]], <8 x i32> poison, <4 x i32> <i32 0, i32 2, i32 4, i32 6>
86// SPIRV-NEXT: [[SHUF2:%.*]] = shufflevector <8 x i32> [[CAST1]], <8 x i32> poison, <4 x i32> <i32 1, i32 3, i32 5, i32 7>
87uint4 test_vector4(double4 D) {
88  uint4 A, B;
89  asuint(D, A, B);
90  return A + B;
91}
92