| ; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z14 -O3 | FileCheck %s |
| ; |
| ; Check that CodeGenPrepare transforms these functions to use |
| ; uadd.with.overflow / usub.with.overflow intrinsics so that the compare |
| ; instruction is eliminated. |
| |
| define i32 @uaddo_32(i32 %arg) { |
| ; CHECK-LABEL: uaddo_32: |
| ; CHECK: alhsik %r0, %r2, -1 |
| ; CHECK: locrnle %r2, %r0 |
| ; CHECK: br %r14 |
| |
| bb: |
| %tmp10 = icmp ne i32 %arg, 0 |
| %tmp11 = add nsw i32 %arg, -1 |
| %tmp12 = select i1 %tmp10, i32 %tmp11, i32 %arg |
| ret i32 %tmp12 |
| } |
| |
| define i64 @uaddo_64(i64 %arg) { |
| ; CHECK-LABEL: uaddo_64: |
| ; CHECK: alghsik %r0, %r2, -1 |
| ; CHECK: locgrnle %r2, %r0 |
| ; CHECK: br %r14 |
| bb: |
| %tmp10 = icmp ne i64 %arg, 0 |
| %tmp11 = add nsw i64 %arg, -1 |
| %tmp12 = select i1 %tmp10, i64 %tmp11, i64 %arg |
| ret i64 %tmp12 |
| } |
| |
| define i32 @usubo_32(i32 %arg) { |
| ; CHECK-LABEL: usubo_32: |
| ; CHECK: alhsik %r0, %r2, -1 |
| ; CHECK: locrle %r2, %r0 |
| ; CHECK: br %r14 |
| bb: |
| %tmp10 = icmp eq i32 %arg, 0 |
| %tmp11 = sub nsw i32 %arg, 1 |
| %tmp12 = select i1 %tmp10, i32 %tmp11, i32 %arg |
| ret i32 %tmp12 |
| } |
| |
| define i64 @usubo_64(i64 %arg) { |
| ; CHECK-LABEL: usubo_64: |
| ; CHECK: alghsik %r0, %r2, -1 |
| ; CHECK: locgrle %r2, %r0 |
| ; CHECK: br %r14 |
| bb: |
| %tmp10 = icmp eq i64 %arg, 0 |
| %tmp11 = sub nsw i64 %arg, 1 |
| %tmp12 = select i1 %tmp10, i64 %tmp11, i64 %arg |
| ret i64 %tmp12 |
| } |
| |
| define i32 @optbranch_32(i32 %Arg) { |
| ; CHECK-LABEL: optbranch_32: |
| ; CHECK: alhsik %r2, %r2, 1 |
| ; CHECK-NEXT: bler %r14 |
| ; CHECK-NEXT: .LBB4_1: |
| ; CHECK-NEXT: lhi %r2, -1 |
| ; CHECK-NEXT: br %r14 |
| bb: |
| %i1 = icmp eq i32 %Arg, -1 |
| br i1 %i1, label %bb2, label %bb3 |
| |
| bb2: |
| ret i32 -1 |
| |
| bb3: |
| %i4 = add nuw i32 %Arg, 1 |
| ret i32 %i4 |
| } |
| |
| define i64 @optbranch_64(i64 %Arg) { |
| ; CHECK-LABEL: optbranch_64: |
| ; CHECK: alghsik %r2, %r2, 1 |
| ; CHECK-NEXT: bler %r14 |
| ; CHECK-NEXT: .LBB5_1: |
| ; CHECK-NEXT: lghi %r2, -1 |
| ; CHECK-NEXT: br %r14 |
| bb: |
| %i1 = icmp eq i64 %Arg, -1 |
| br i1 %i1, label %bb2, label %bb3 |
| |
| bb2: |
| ret i64 -1 |
| |
| bb3: |
| %i4 = add nuw i64 %Arg, 1 |
| ret i64 %i4 |
| } |