diff --git a/llvm/test/Transforms/InstSimplify/select.ll b/llvm/test/Transforms/InstSimplify/select.ll --- a/llvm/test/Transforms/InstSimplify/select.ll +++ b/llvm/test/Transforms/InstSimplify/select.ll @@ -1438,3 +1438,69 @@ exit: ret ptr %sel } + +define i8 @select_sub_cmp(i8 %0, i8 %1) { +; CHECK-LABEL: @select_sub_cmp( +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq i8 [[TMP1:%.*]], [[TMP0:%.*]] +; CHECK-NEXT: [[TMP4:%.*]] = sub nsw i8 [[TMP1]], [[TMP0]] +; CHECK-NEXT: [[TMP5:%.*]] = select i1 [[TMP3]], i8 0, i8 [[TMP4]] +; CHECK-NEXT: ret i8 [[TMP5]] +; + %3 = icmp eq i8 %1, %0 + %4 = sub nsw i8 %1, %0 + %5 = select i1 %3, i8 0, i8 %4 + ret i8 %5 +} + +define <2 x i8> @select_sub_cmp_vec(<2 x i8> %0, <2 x i8> %1) { +; CHECK-LABEL: @select_sub_cmp_vec( +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq <2 x i8> [[TMP1:%.*]], [[TMP0:%.*]] +; CHECK-NEXT: [[TMP4:%.*]] = sub nsw <2 x i8> [[TMP1]], [[TMP0]] +; CHECK-NEXT: [[TMP5:%.*]] = select <2 x i1> [[TMP3]], <2 x i8> zeroinitializer, <2 x i8> [[TMP4]] +; CHECK-NEXT: ret <2 x i8> [[TMP5]] +; + %3 = icmp eq <2 x i8> %1, %0 + %4 = sub nsw <2 x i8> %1, %0 + %5 = select <2 x i1> %3, <2 x i8> , <2 x i8> %4 + ret <2 x i8> %5 +} + +define i8 @select_sub_cmp_swap(i8 %0, i8 %1) { +; CHECK-LABEL: @select_sub_cmp_swap( +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq i8 [[TMP1:%.*]], [[TMP0:%.*]] +; CHECK-NEXT: [[TMP4:%.*]] = sub nsw i8 [[TMP0]], [[TMP1]] +; CHECK-NEXT: [[TMP5:%.*]] = select i1 [[TMP3]], i8 0, i8 [[TMP4]] +; CHECK-NEXT: ret i8 [[TMP5]] +; + %3 = icmp eq i8 %1, %0 + %4 = sub nsw i8 %0, %1 + %5 = select i1 %3, i8 0, i8 %4 + ret i8 %5 +} + +define <2 x i8> @select_sub_cmp_vec_swap(<2 x i8> %0, <2 x i8> %1) { +; CHECK-LABEL: @select_sub_cmp_vec_swap( +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq <2 x i8> [[TMP1:%.*]], [[TMP0:%.*]] +; CHECK-NEXT: [[TMP4:%.*]] = sub nsw <2 x i8> [[TMP0]], [[TMP1]] +; CHECK-NEXT: [[TMP5:%.*]] = select <2 x i1> [[TMP3]], <2 x i8> zeroinitializer, <2 x i8> [[TMP4]] +; CHECK-NEXT: ret <2 x i8> [[TMP5]] +; + %3 = icmp eq <2 x i8> %1, %0 + %4 = sub nsw <2 x i8> %0, %1 + %5 = select <2 x i1> %3, <2 x i8> , <2 x i8> %4 + ret <2 x i8> %5 +} + +; negative test +define i8 @select_sub_cmp_nonzero(i8 %0, i8 %1) { +; CHECK-LABEL: @select_sub_cmp_nonzero( +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq i8 [[TMP1:%.*]], [[TMP0:%.*]] +; CHECK-NEXT: [[TMP4:%.*]] = sub nsw i8 [[TMP1]], [[TMP0]] +; CHECK-NEXT: [[TMP5:%.*]] = select i1 [[TMP3]], i8 42, i8 [[TMP4]] +; CHECK-NEXT: ret i8 [[TMP5]] +; + %3 = icmp eq i8 %1, %0 + %4 = sub nsw i8 %1, %0 + %5 = select i1 %3, i8 42, i8 %4 + ret i8 %5 +}