diff options
author | Matt Arsenault <Matthew.Arsenault@amd.com> | 2013-11-04 20:36:06 +0000 |
---|---|---|
committer | Matt Arsenault <Matthew.Arsenault@amd.com> | 2013-11-04 20:36:06 +0000 |
commit | eba6d384489be4e56718186aa7ed7e484df24613 (patch) | |
tree | 15236fcdfdc1c73f6028c0838d762a7054a9d34e /test/Transforms/InstCombine | |
parent | da666ab229c0987090c4f3d60ce7f46df8c50360 (diff) | |
download | llvm-eba6d384489be4e56718186aa7ed7e484df24613.tar.gz llvm-eba6d384489be4e56718186aa7ed7e484df24613.tar.bz2 llvm-eba6d384489be4e56718186aa7ed7e484df24613.tar.xz |
Scalarize select vector arguments when extracted.
When the elements are extracted from a select on vectors
or a vector select, do the select on the extracted scalars
from the input if there is only one use.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@194013 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'test/Transforms/InstCombine')
-rw-r--r-- | test/Transforms/InstCombine/select-extractelement.ll | 102 |
1 files changed, 102 insertions, 0 deletions
diff --git a/test/Transforms/InstCombine/select-extractelement.ll b/test/Transforms/InstCombine/select-extractelement.ll new file mode 100644 index 0000000000..e7ea851d92 --- /dev/null +++ b/test/Transforms/InstCombine/select-extractelement.ll @@ -0,0 +1,102 @@ +; 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" } |