| ; RUN: opt -S -instcombine < %s | FileCheck %s |
| |
| declare void @v4float_user(<4 x float>) #0 |
| |
| |
| |
| define float @extract_one_select(<4 x float> %a, <4 x float> %b, i32 %c) #0 { |
| ; CHECK-LABEL: @extract_one_select( |
| ; CHECK-NOT: select i1 {{.*}}, <4 x float> |
| %cmp = icmp ne i32 %c, 0 |
| %sel = select i1 %cmp, <4 x float> %a, <4 x float> %b |
| %extract = extractelement <4 x float> %sel, i32 2 |
| ret float %extract |
| } |
| |
| ; Multiple extractelements |
| define <2 x float> @extract_two_select(<4 x float> %a, <4 x float> %b, i32 %c) #0 { |
| ; CHECK-LABEL: @extract_two_select( |
| ; CHECK: select i1 {{.*}}, <4 x float> |
| %cmp = icmp ne i32 %c, 0 |
| %sel = select i1 %cmp, <4 x float> %a, <4 x float> %b |
| %extract1 = extractelement <4 x float> %sel, i32 1 |
| %extract2 = extractelement <4 x float> %sel, i32 2 |
| %build1 = insertelement <2 x float> undef, float %extract1, i32 0 |
| %build2 = insertelement <2 x float> %build1, float %extract2, i32 1 |
| ret <2 x float> %build2 |
| } |
| |
| ; Select has an extra non-extractelement user, don't change it |
| define float @extract_one_select_user(<4 x float> %a, <4 x float> %b, i32 %c) #0 { |
| ; CHECK-LABEL: @extract_one_select_user( |
| ; CHECK: select i1 {{.*}}, <4 x float> |
| %cmp = icmp ne i32 %c, 0 |
| %sel = select i1 %cmp, <4 x float> %a, <4 x float> %b |
| %extract = extractelement <4 x float> %sel, i32 2 |
| call void @v4float_user(<4 x float> %sel) |
| ret float %extract |
| } |
| |
| define float @extract_one_vselect_user(<4 x float> %a, <4 x float> %b, <4 x i32> %c) #0 { |
| ; CHECK-LABEL: @extract_one_vselect_user( |
| ; CHECK: select <4 x i1> {{.*}}, <4 x float> |
| %cmp = icmp ne <4 x i32> %c, zeroinitializer |
| %sel = select <4 x i1> %cmp, <4 x float> %a, <4 x float> %b |
| %extract = extractelement <4 x float> %sel, i32 2 |
| call void @v4float_user(<4 x float> %sel) |
| ret float %extract |
| } |
| |
| ; Extract from a vector select |
| define float @extract_one_vselect(<4 x float> %a, <4 x float> %b, <4 x i32> %c) #0 { |
| ; CHECK-LABEL: @extract_one_vselect( |
| ; CHECK-NOT: select <4 x i1> |
| %cmp = icmp ne <4 x i32> %c, zeroinitializer |
| %select = select <4 x i1> %cmp, <4 x float> %a, <4 x float> %b |
| %extract = extractelement <4 x float> %select, i32 0 |
| ret float %extract |
| } |
| |
| ; Multiple extractelements from a vector select |
| define <2 x float> @extract_two_vselect(<4 x float> %a, <4 x float> %b, <4 x i32> %c) #0 { |
| ; CHECK-LABEL: @extract_two_vselect( |
| ; CHECK-NOT: select i1 {{.*}}, <4 x float> |
| %cmp = icmp ne <4 x i32> %c, zeroinitializer |
| %sel = select <4 x i1> %cmp, <4 x float> %a, <4 x float> %b |
| %extract1 = extractelement <4 x float> %sel, i32 1 |
| %extract2 = extractelement <4 x float> %sel, i32 2 |
| %build1 = insertelement <2 x float> undef, float %extract1, i32 0 |
| %build2 = insertelement <2 x float> %build1, float %extract2, i32 1 |
| ret <2 x float> %build2 |
| } |
| |
| ; All the vector selects should be decomposed into scalar selects |
| ; Test multiple extractelements |
| define <4 x float> @simple_vector_select(<4 x float> %a, <4 x float> %b, <4 x i32> %c) #0 { |
| ; CHECK-LABEL: @simple_vector_select( |
| ; CHECK-NOT: select i1 {{.*}}, <4 x float> |
| entry: |
| %0 = extractelement <4 x i32> %c, i32 0 |
| %tobool = icmp ne i32 %0, 0 |
| %a.sink = select i1 %tobool, <4 x float> %a, <4 x float> %b |
| %1 = extractelement <4 x float> %a.sink, i32 0 |
| %2 = insertelement <4 x float> undef, float %1, i32 0 |
| %3 = extractelement <4 x i32> %c, i32 1 |
| %tobool1 = icmp ne i32 %3, 0 |
| %a.sink1 = select i1 %tobool1, <4 x float> %a, <4 x float> %b |
| %4 = extractelement <4 x float> %a.sink1, i32 1 |
| %5 = insertelement <4 x float> %2, float %4, i32 1 |
| %6 = extractelement <4 x i32> %c, i32 2 |
| %tobool6 = icmp ne i32 %6, 0 |
| %a.sink2 = select i1 %tobool6, <4 x float> %a, <4 x float> %b |
| %7 = extractelement <4 x float> %a.sink2, i32 2 |
| %8 = insertelement <4 x float> %5, float %7, i32 2 |
| %9 = extractelement <4 x i32> %c, i32 3 |
| %tobool11 = icmp ne i32 %9, 0 |
| %a.sink3 = select i1 %tobool11, <4 x float> %a, <4 x float> %b |
| %10 = extractelement <4 x float> %a.sink3, i32 3 |
| %11 = insertelement <4 x float> %8, float %10, i32 3 |
| ret <4 x float> %11 |
| } |
| |
| attributes #0 = { nounwind ssp uwtable "less-precise-fpmad"="false" "no-frame-pointer-elim"="true" "no-frame-pointer-elim-non-leaf"="true" "no-infs-fp-math"="false" "no-nans-fp-math"="false" "stack-protector-buffer-size"="8" "unsafe-fp-math"="false" "use-soft-float"="false" } |