130 lines · plain
1// RUN: %clang_cc1 -finclude-default-header -x hlsl -triple \2// RUN: dxil-pc-shadermodel6.3-library %s -emit-llvm -disable-llvm-passes \3// RUN: -o - | FileCheck %s --check-prefixes=CHECK4 5// CHECK-LABEL: test_select_bool_int6// CHECK: [[SELECT:%.*]] = select i1 {{%.*}}, i32 {{%.*}}, i32 {{%.*}}7// CHECK: ret i32 [[SELECT]]8int test_select_bool_int(bool cond0, int tVal, int fVal) {9 return select<int>(cond0, tVal, fVal);10}11 12struct S { int a; };13// CHECK-LABEL: test_select_infer_struct14// CHECK: [[TRUE_VAL:%.*]] = load %struct.S, ptr {{%.*}}, align 115// CHECK: [[FALSE_VAL:%.*]] = load %struct.S, ptr {{%.*}}, align 116// CHECK: [[SELECT:%.*]] = select i1 {{%.*}}, %struct.S [[TRUE_VAL]], %struct.S [[FALSE_VAL]]17// CHECK: store %struct.S [[SELECT]], ptr {{%.*}}, align 118// CHECK: ret void19struct S test_select_infer_struct(bool cond0, struct S tVal, struct S fVal) {20 return select(cond0, tVal, fVal);21}22 23// CHECK-LABEL: test_select_bool_vector24// CHECK: [[SELECT:%.*]] = select i1 {{%.*}}, <2 x i32> {{%.*}}, <2 x i32> {{%.*}}25// CHECK: ret <2 x i32> [[SELECT]]26int2 test_select_bool_vector(bool cond0, int2 tVal, int2 fVal) {27 return select<int2>(cond0, tVal, fVal);28}29 30// CHECK-LABEL: test_select_vector_131// CHECK: [[SELECT:%.*]] = select i1 {{%.*}}, <1 x i32> {{%.*}}, <1 x i32> {{%.*}}32// CHECK: ret <1 x i32> [[SELECT]]33int1 test_select_vector_1(bool1 cond0, int1 tVals, int1 fVals) {34 return select(cond0, tVals, fVals);35}36 37// CHECK-LABEL: test_select_vector_238// CHECK: [[SELECT:%.*]] = select <2 x i1> {{%.*}}, <2 x i32> {{%.*}}, <2 x i32> {{%.*}}39// CHECK: ret <2 x i32> [[SELECT]]40int2 test_select_vector_2(bool2 cond0, int2 tVals, int2 fVals) {41 return select(cond0, tVals, fVals);42}43 44// CHECK-LABEL: test_select_vector_345// CHECK: [[SELECT:%.*]] = select <3 x i1> {{%.*}}, <3 x i32> {{%.*}}, <3 x i32> {{%.*}}46// CHECK: ret <3 x i32> [[SELECT]]47int3 test_select_vector_3(bool3 cond0, int3 tVals, int3 fVals) {48 return select(cond0, tVals, fVals);49}50 51// CHECK-LABEL: test_select_vector_452// CHECK: [[SELECT:%.*]] = select <4 x i1> {{%.*}}, <4 x i32> {{%.*}}, <4 x i32> {{%.*}}53// CHECK: ret <4 x i32> [[SELECT]]54int4 test_select_vector_4(bool4 cond0, int4 tVals, int4 fVals) {55 return select(cond0, tVals, fVals);56}57 58// CHECK-LABEL: test_select_vector_scalar_vector59// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 060// CHECK: [[SPLAT1:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC1]], <4 x i32> poison, <4 x i32> zeroinitializer61// CHECK: [[SELECT:%.*]] = select <4 x i1> {{%.*}}, <4 x i32> [[SPLAT1]], <4 x i32> {{%.*}}62// CHECK: ret <4 x i32> [[SELECT]]63int4 test_select_vector_scalar_vector(bool4 cond0, int tVal, int4 fVals) {64 return select(cond0, tVal, fVals);65}66 67// CHECK-LABEL: test_select_vector_vector_scalar68// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 069// CHECK: [[SPLAT1:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC1]], <4 x i32> poison, <4 x i32> zeroinitializer70// CHECK: [[SELECT:%.*]] = select <4 x i1> {{%.*}}, <4 x i32> {{%.*}}, <4 x i32> [[SPLAT1]]71// CHECK: ret <4 x i32> [[SELECT]]72int4 test_select_vector_vector_scalar(bool4 cond0, int4 tVals, int fVal) {73 return select(cond0, tVals, fVal);74}75 76// CHECK-LABEL: test_select_vector_scalar_scalar77// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 078// CHECK: [[SPLAT1:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC1]], <4 x i32> poison, <4 x i32> zeroinitializer79// CHECK: [[SPLAT_SRC2:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 080// CHECK: [[SPLAT2:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC2]], <4 x i32> poison, <4 x i32> zeroinitializer81// CHECK: [[SELECT:%.*]] = select <4 x i1> {{%.*}}, <4 x i32> [[SPLAT1]], <4 x i32> [[SPLAT2]]82// CHECK: ret <4 x i32> [[SELECT]]83int4 test_select_vector_scalar_scalar(bool4 cond0, int tVal, int fVal) {84 return select(cond0, tVal, fVal);85}86 87// CHECK-LABEL: test_select_nonbool_cond_vector_488// CHECK: [[TMP0:%.*]] = load <4 x i32>, ptr %cond0.addr, align 1689// CHECK: [[TOBOOL:%.*]] = icmp ne <4 x i32> [[TMP0]], zeroinitializer90// CHECK: [[SELECT:%.*]] = select <4 x i1> [[TOBOOL]], <4 x i1> {{%.*}}, <4 x i1> {{%.*}}91// CHECK: ret <4 x i1> [[SELECT]]92bool4 test_select_nonbool_cond_vector_4(int4 cond0, bool4 tVal, bool4 fVal) {93 return select(cond0, tVal, fVal);94}95 96// CHECK-LABEL: test_select_nonbool_cond_vector_scalar_vector97// CHECK: [[TMP0:%.*]] = load <3 x i32>, ptr %cond0.addr, align 1698// CHECK: [[TOBOOL:%.*]] = icmp ne <3 x i32> [[TMP0]], zeroinitializer99// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <3 x i32> poison, i32 {{%.*}}, i64 0100// CHECK: [[SPLAT1:%.*]] = shufflevector <3 x i32> [[SPLAT_SRC1]], <3 x i32> poison, <3 x i32> zeroinitializer101// CHECK: [[SELECT:%.*]] = select <3 x i1> [[TOBOOL]], <3 x i32> [[SPLAT1]], <3 x i32> {{%.*}}102// CHECK: ret <3 x i32> [[SELECT]]103int3 test_select_nonbool_cond_vector_scalar_vector(int3 cond0, int tVal, int3 fVal) {104 return select(cond0, tVal, fVal);105}106 107// CHECK-LABEL: test_select_nonbool_cond_vector_vector_scalar108// CHECK: [[TMP0:%.*]] = load <2 x i32>, ptr %cond0.addr, align 8109// CHECK: [[TOBOOL:%.*]] = icmp ne <2 x i32> [[TMP0]], zeroinitializer110// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <2 x i32> poison, i32 {{%.*}}, i64 0111// CHECK: [[SPLAT1:%.*]] = shufflevector <2 x i32> [[SPLAT_SRC1]], <2 x i32> poison, <2 x i32> zeroinitializer112// CHECK: [[SELECT:%.*]] = select <2 x i1> [[TOBOOL]], <2 x i32> {{%.*}}, <2 x i32> [[SPLAT1]]113// CHECK: ret <2 x i32> [[SELECT]]114int2 test_select_nonbool_cond_vector_vector_scalar(int2 cond0, int2 tVal, int fVal) {115 return select(cond0, tVal, fVal);116}117 118// CHECK-LABEL: test_select_nonbool_cond_vector_scalar_scalar119// CHECK: [[TMP0:%.*]] = load <4 x i32>, ptr %cond0.addr, align 16120// CHECK: [[TOBOOL:%.*]] = icmp ne <4 x i32> [[TMP0]], zeroinitializer121// CHECK: [[SPLAT_SRC1:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 0122// CHECK: [[SPLAT1:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC1]], <4 x i32> poison, <4 x i32> zeroinitializer123// CHECK: [[SPLAT_SRC2:%.*]] = insertelement <4 x i32> poison, i32 {{%.*}}, i64 0124// CHECK: [[SPLAT2:%.*]] = shufflevector <4 x i32> [[SPLAT_SRC2]], <4 x i32> poison, <4 x i32> zeroinitializer125// CHECK: [[SELECT:%.*]] = select <4 x i1> [[TOBOOL]], <4 x i32> [[SPLAT1]], <4 x i32> [[SPLAT2]]126// CHECK: ret <4 x i32> [[SELECT]]127int4 test_select_nonbool_cond_vector_scalar_scalar(int4 cond0, int tVal, int fVal) {128 return select(cond0, tVal, fVal);129}130