clang/test/CodeGenHLSL/builtins/splitdouble.hlsl

   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
  17 uint 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
  36 uint1 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>
  53 uint2 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>
  70 uint3 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>
  87 uint4 test_vector4(double4 D) {
  88   uint4 A, B;
  89   asuint(D, A, B);
  90   return A + B;
  91 }