393 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py2; RUN: llc < %s | FileCheck --check-prefixes=CHECK,CHECK-SD %s3; RUN: llc < %s -global-isel | FileCheck --check-prefixes=CHECK,CHECK-GI %s4 5target triple = "aarch64-none-linux-gnu"6 7define i32 @test_basic_i32(i32 %a, i32 %b, i32 %x, i32 %y) {8; CHECK-SD-LABEL: test_basic_i32:9; CHECK-SD: // %bb.0:10; CHECK-SD-NEXT: cmp w0, w111; CHECK-SD-NEXT: sbc w0, w2, w312; CHECK-SD-NEXT: ret13;14; CHECK-GI-LABEL: test_basic_i32:15; CHECK-GI: // %bb.0:16; CHECK-GI-NEXT: cmp w0, w117; CHECK-GI-NEXT: sub w9, w2, w318; CHECK-GI-NEXT: cset w8, lo19; CHECK-GI-NEXT: sub w0, w9, w820; CHECK-GI-NEXT: ret21 %cc = icmp ult i32 %a, %b22 %carry = zext i1 %cc to i3223 %sub = sub i32 %x, %y24 %res = sub i32 %sub, %carry25 ret i32 %res26}27 28define i64 @test_basic_i64(i64 %a, i64 %b, i64 %x, i64 %y) {29; CHECK-SD-LABEL: test_basic_i64:30; CHECK-SD: // %bb.0:31; CHECK-SD-NEXT: cmp x0, x132; CHECK-SD-NEXT: sbc x0, x2, x333; CHECK-SD-NEXT: ret34;35; CHECK-GI-LABEL: test_basic_i64:36; CHECK-GI: // %bb.0:37; CHECK-GI-NEXT: cmp x0, x138; CHECK-GI-NEXT: sub x9, x2, x339; CHECK-GI-NEXT: cset w8, lo40; CHECK-GI-NEXT: sub x0, x9, x841; CHECK-GI-NEXT: ret42 %cc = icmp ult i64 %a, %b43 %carry = zext i1 %cc to i6444 %sub = sub i64 %x, %y45 %res = sub i64 %sub, %carry46 ret i64 %res47}48 49define i64 @test_mixed_i32_i64(i32 %a, i32 %b, i64 %x, i64 %y) {50; CHECK-SD-LABEL: test_mixed_i32_i64:51; CHECK-SD: // %bb.0:52; CHECK-SD-NEXT: cmp w0, w153; CHECK-SD-NEXT: sbc x0, x2, x354; CHECK-SD-NEXT: ret55;56; CHECK-GI-LABEL: test_mixed_i32_i64:57; CHECK-GI: // %bb.0:58; CHECK-GI-NEXT: cmp w0, w159; CHECK-GI-NEXT: sub x9, x2, x360; CHECK-GI-NEXT: cset w8, lo61; CHECK-GI-NEXT: sub x0, x9, x862; CHECK-GI-NEXT: ret63 %cc = icmp ult i32 %a, %b64 %carry = zext i1 %cc to i6465 %sub = sub i64 %x, %y66 %res = sub i64 %sub, %carry67 ret i64 %res68}69 70define i32 @test_mixed_i64_i32(i64 %a, i64 %b, i32 %x, i32 %y) {71; CHECK-SD-LABEL: test_mixed_i64_i32:72; CHECK-SD: // %bb.0:73; CHECK-SD-NEXT: cmp x0, x174; CHECK-SD-NEXT: sbc w0, w2, w375; CHECK-SD-NEXT: ret76;77; CHECK-GI-LABEL: test_mixed_i64_i32:78; CHECK-GI: // %bb.0:79; CHECK-GI-NEXT: cmp x0, x180; CHECK-GI-NEXT: sub w9, w2, w381; CHECK-GI-NEXT: cset w8, lo82; CHECK-GI-NEXT: sub w0, w9, w883; CHECK-GI-NEXT: ret84 %cc = icmp ult i64 %a, %b85 %carry = zext i1 %cc to i3286 %sub = sub i32 %x, %y87 %res = sub i32 %sub, %carry88 ret i32 %res89}90 91define i32 @test_only_borrow(i32 %a, i32 %b, i32 %x) {92; CHECK-SD-LABEL: test_only_borrow:93; CHECK-SD: // %bb.0:94; CHECK-SD-NEXT: cmp w0, w195; CHECK-SD-NEXT: sbc w0, w2, wzr96; CHECK-SD-NEXT: ret97;98; CHECK-GI-LABEL: test_only_borrow:99; CHECK-GI: // %bb.0:100; CHECK-GI-NEXT: cmp w0, w1101; CHECK-GI-NEXT: cset w8, lo102; CHECK-GI-NEXT: sub w0, w2, w8103; CHECK-GI-NEXT: ret104 %cc = icmp ult i32 %a, %b105 %carry = zext i1 %cc to i32106 %res = sub i32 %x, %carry107 ret i32 %res108}109 110define i32 @test_sext_add(i32 %a, i32 %b, i32 %x, i32 %y) {111; CHECK-SD-LABEL: test_sext_add:112; CHECK-SD: // %bb.0:113; CHECK-SD-NEXT: cmp w0, w1114; CHECK-SD-NEXT: sbc w0, w2, w3115; CHECK-SD-NEXT: ret116;117; CHECK-GI-LABEL: test_sext_add:118; CHECK-GI: // %bb.0:119; CHECK-GI-NEXT: cmp w0, w1120; CHECK-GI-NEXT: sub w9, w2, w3121; CHECK-GI-NEXT: cset w8, lo122; CHECK-GI-NEXT: sbfx w8, w8, #0, #1123; CHECK-GI-NEXT: add w0, w9, w8124; CHECK-GI-NEXT: ret125 %cc = icmp ult i32 %a, %b126 %carry = sext i1 %cc to i32127 %sub = sub i32 %x, %y128 %res = add i32 %sub, %carry129 ret i32 %res130}131 132; FIXME: This case could be supported with reversed operands to the CMP.133define i32 @test_ugt(i32 %a, i32 %b, i32 %x, i32 %y) {134; CHECK-SD-LABEL: test_ugt:135; CHECK-SD: // %bb.0:136; CHECK-SD-NEXT: cmp w0, w1137; CHECK-SD-NEXT: sub w8, w2, w3138; CHECK-SD-NEXT: cset w9, hi139; CHECK-SD-NEXT: sub w0, w8, w9140; CHECK-SD-NEXT: ret141;142; CHECK-GI-LABEL: test_ugt:143; CHECK-GI: // %bb.0:144; CHECK-GI-NEXT: cmp w0, w1145; CHECK-GI-NEXT: sub w9, w2, w3146; CHECK-GI-NEXT: cset w8, hi147; CHECK-GI-NEXT: sub w0, w9, w8148; CHECK-GI-NEXT: ret149 %cc = icmp ugt i32 %a, %b150 %carry = zext i1 %cc to i32151 %sub = sub i32 %x, %y152 %res = sub i32 %sub, %carry153 ret i32 %res154}155 156define i32 @test_unsupported_cc_slt(i32 %a, i32 %b, i32 %x, i32 %y) {157; CHECK-SD-LABEL: test_unsupported_cc_slt:158; CHECK-SD: // %bb.0:159; CHECK-SD-NEXT: cmp w0, w1160; CHECK-SD-NEXT: sub w8, w2, w3161; CHECK-SD-NEXT: cset w9, lt162; CHECK-SD-NEXT: sub w0, w8, w9163; CHECK-SD-NEXT: ret164;165; CHECK-GI-LABEL: test_unsupported_cc_slt:166; CHECK-GI: // %bb.0:167; CHECK-GI-NEXT: cmp w0, w1168; CHECK-GI-NEXT: sub w9, w2, w3169; CHECK-GI-NEXT: cset w8, lt170; CHECK-GI-NEXT: sub w0, w9, w8171; CHECK-GI-NEXT: ret172 %cc = icmp slt i32 %a, %b173 %carry = zext i1 %cc to i32174 %sub = sub i32 %x, %y175 %res = sub i32 %sub, %carry176 ret i32 %res177}178 179define i32 @test_unsupported_cc_sgt(i32 %a, i32 %b, i32 %x, i32 %y) {180; CHECK-SD-LABEL: test_unsupported_cc_sgt:181; CHECK-SD: // %bb.0:182; CHECK-SD-NEXT: cmp w0, w1183; CHECK-SD-NEXT: sub w8, w2, w3184; CHECK-SD-NEXT: cset w9, gt185; CHECK-SD-NEXT: sub w0, w8, w9186; CHECK-SD-NEXT: ret187;188; CHECK-GI-LABEL: test_unsupported_cc_sgt:189; CHECK-GI: // %bb.0:190; CHECK-GI-NEXT: cmp w0, w1191; CHECK-GI-NEXT: sub w9, w2, w3192; CHECK-GI-NEXT: cset w8, gt193; CHECK-GI-NEXT: sub w0, w9, w8194; CHECK-GI-NEXT: ret195 %cc = icmp sgt i32 %a, %b196 %carry = zext i1 %cc to i32197 %sub = sub i32 %x, %y198 %res = sub i32 %sub, %carry199 ret i32 %res200}201 202define i32 @test_multiple_setcc_uses(i32 %a, i32 %b, i32 %x) {203; CHECK-SD-LABEL: test_multiple_setcc_uses:204; CHECK-SD: // %bb.0:205; CHECK-SD-NEXT: stp x30, x19, [sp, #-16]! // 16-byte Folded Spill206; CHECK-SD-NEXT: .cfi_def_cfa_offset 16207; CHECK-SD-NEXT: .cfi_offset w19, -8208; CHECK-SD-NEXT: .cfi_offset w30, -16209; CHECK-SD-NEXT: cmp w0, w1210; CHECK-SD-NEXT: cset w0, lo211; CHECK-SD-NEXT: sub w19, w2, w0212; CHECK-SD-NEXT: bl use213; CHECK-SD-NEXT: mov w0, w19214; CHECK-SD-NEXT: ldp x30, x19, [sp], #16 // 16-byte Folded Reload215; CHECK-SD-NEXT: ret216;217; CHECK-GI-LABEL: test_multiple_setcc_uses:218; CHECK-GI: // %bb.0:219; CHECK-GI-NEXT: str x30, [sp, #-32]! // 8-byte Folded Spill220; CHECK-GI-NEXT: stp x20, x19, [sp, #16] // 16-byte Folded Spill221; CHECK-GI-NEXT: .cfi_def_cfa_offset 32222; CHECK-GI-NEXT: .cfi_offset w19, -8223; CHECK-GI-NEXT: .cfi_offset w20, -16224; CHECK-GI-NEXT: .cfi_offset w30, -32225; CHECK-GI-NEXT: cmp w0, w1226; CHECK-GI-NEXT: mov w19, w2227; CHECK-GI-NEXT: cset w20, lo228; CHECK-GI-NEXT: mov w0, w20229; CHECK-GI-NEXT: bl use230; CHECK-GI-NEXT: sub w0, w19, w20231; CHECK-GI-NEXT: ldp x20, x19, [sp, #16] // 16-byte Folded Reload232; CHECK-GI-NEXT: ldr x30, [sp], #32 // 8-byte Folded Reload233; CHECK-GI-NEXT: ret234 %cc = icmp ult i32 %a, %b235 %carry = zext i1 %cc to i32236 %res = sub i32 %x, %carry237 tail call void @use(i1 %cc)238 ret i32 %res239}240 241define i32 @test_multiple_carry_uses(i32 %a, i32 %b, i32 %x) {242; CHECK-SD-LABEL: test_multiple_carry_uses:243; CHECK-SD: // %bb.0:244; CHECK-SD-NEXT: stp x30, x19, [sp, #-16]! // 16-byte Folded Spill245; CHECK-SD-NEXT: .cfi_def_cfa_offset 16246; CHECK-SD-NEXT: .cfi_offset w19, -8247; CHECK-SD-NEXT: .cfi_offset w30, -16248; CHECK-SD-NEXT: cmp w0, w1249; CHECK-SD-NEXT: cset w0, lo250; CHECK-SD-NEXT: sub w19, w2, w0251; CHECK-SD-NEXT: bl use252; CHECK-SD-NEXT: mov w0, w19253; CHECK-SD-NEXT: ldp x30, x19, [sp], #16 // 16-byte Folded Reload254; CHECK-SD-NEXT: ret255;256; CHECK-GI-LABEL: test_multiple_carry_uses:257; CHECK-GI: // %bb.0:258; CHECK-GI-NEXT: str x30, [sp, #-32]! // 8-byte Folded Spill259; CHECK-GI-NEXT: stp x20, x19, [sp, #16] // 16-byte Folded Spill260; CHECK-GI-NEXT: .cfi_def_cfa_offset 32261; CHECK-GI-NEXT: .cfi_offset w19, -8262; CHECK-GI-NEXT: .cfi_offset w20, -16263; CHECK-GI-NEXT: .cfi_offset w30, -32264; CHECK-GI-NEXT: cmp w0, w1265; CHECK-GI-NEXT: mov w19, w2266; CHECK-GI-NEXT: cset w20, lo267; CHECK-GI-NEXT: mov w0, w20268; CHECK-GI-NEXT: bl use269; CHECK-GI-NEXT: sub w0, w19, w20270; CHECK-GI-NEXT: ldp x20, x19, [sp, #16] // 16-byte Folded Reload271; CHECK-GI-NEXT: ldr x30, [sp], #32 // 8-byte Folded Reload272; CHECK-GI-NEXT: ret273 %cc = icmp ult i32 %a, %b274 %carry = zext i1 %cc to i32275 %res = sub i32 %x, %carry276 tail call void @use(i32 %carry)277 ret i32 %res278}279 280define i32 @test_multiple_sub_uses(i32 %a, i32 %b, i32 %x, i32 %y) {281; CHECK-SD-LABEL: test_multiple_sub_uses:282; CHECK-SD: // %bb.0:283; CHECK-SD-NEXT: stp x30, x19, [sp, #-16]! // 16-byte Folded Spill284; CHECK-SD-NEXT: .cfi_def_cfa_offset 16285; CHECK-SD-NEXT: .cfi_offset w19, -8286; CHECK-SD-NEXT: .cfi_offset w30, -16287; CHECK-SD-NEXT: sub w8, w2, w3288; CHECK-SD-NEXT: cmp w0, w1289; CHECK-SD-NEXT: mov w0, w8290; CHECK-SD-NEXT: sbc w19, w2, w3291; CHECK-SD-NEXT: bl use292; CHECK-SD-NEXT: mov w0, w19293; CHECK-SD-NEXT: ldp x30, x19, [sp], #16 // 16-byte Folded Reload294; CHECK-SD-NEXT: ret295;296; CHECK-GI-LABEL: test_multiple_sub_uses:297; CHECK-GI: // %bb.0:298; CHECK-GI-NEXT: str x30, [sp, #-32]! // 8-byte Folded Spill299; CHECK-GI-NEXT: stp x20, x19, [sp, #16] // 16-byte Folded Spill300; CHECK-GI-NEXT: .cfi_def_cfa_offset 32301; CHECK-GI-NEXT: .cfi_offset w19, -8302; CHECK-GI-NEXT: .cfi_offset w20, -16303; CHECK-GI-NEXT: .cfi_offset w30, -32304; CHECK-GI-NEXT: sub w19, w2, w3305; CHECK-GI-NEXT: cmp w0, w1306; CHECK-GI-NEXT: mov w0, w19307; CHECK-GI-NEXT: cset w20, lo308; CHECK-GI-NEXT: bl use309; CHECK-GI-NEXT: sub w0, w19, w20310; CHECK-GI-NEXT: ldp x20, x19, [sp, #16] // 16-byte Folded Reload311; CHECK-GI-NEXT: ldr x30, [sp], #32 // 8-byte Folded Reload312; CHECK-GI-NEXT: ret313 %cc = icmp ult i32 %a, %b314 %carry = zext i1 %cc to i32315 %sub = sub i32 %x, %y316 %res = sub i32 %sub, %carry317 tail call void @use(i32 %sub)318 ret i32 %res319}320 321define i8 @test_i8(i8 %a, i8 %b, i8 %x, i8 %y) {322; CHECK-SD-LABEL: test_i8:323; CHECK-SD: // %bb.0:324; CHECK-SD-NEXT: and w8, w0, #0xff325; CHECK-SD-NEXT: cmp w8, w1, uxtb326; CHECK-SD-NEXT: sbc w0, w2, w3327; CHECK-SD-NEXT: ret328;329; CHECK-GI-LABEL: test_i8:330; CHECK-GI: // %bb.0:331; CHECK-GI-NEXT: and w8, w0, #0xff332; CHECK-GI-NEXT: sub w9, w2, w3333; CHECK-GI-NEXT: cmp w8, w1, uxtb334; CHECK-GI-NEXT: cset w8, lo335; CHECK-GI-NEXT: sub w0, w9, w8336; CHECK-GI-NEXT: ret337 %cc = icmp ult i8 %a, %b338 %carry = zext i1 %cc to i8339 %sub = sub i8 %x, %y340 %res = sub i8 %sub, %carry341 ret i8 %res342}343 344define i16 @test_i16(i16 %a, i16 %b, i16 %x, i16 %y) {345; CHECK-SD-LABEL: test_i16:346; CHECK-SD: // %bb.0:347; CHECK-SD-NEXT: and w8, w0, #0xffff348; CHECK-SD-NEXT: cmp w8, w1, uxth349; CHECK-SD-NEXT: sbc w0, w2, w3350; CHECK-SD-NEXT: ret351;352; CHECK-GI-LABEL: test_i16:353; CHECK-GI: // %bb.0:354; CHECK-GI-NEXT: and w8, w0, #0xffff355; CHECK-GI-NEXT: sub w9, w2, w3356; CHECK-GI-NEXT: cmp w8, w1, uxth357; CHECK-GI-NEXT: cset w8, lo358; CHECK-GI-NEXT: sub w0, w9, w8359; CHECK-GI-NEXT: ret360 %cc = icmp ult i16 %a, %b361 %carry = zext i1 %cc to i16362 %sub = sub i16 %x, %y363 %res = sub i16 %sub, %carry364 ret i16 %res365}366 367define <4 x i32> @test_v4i32(<4 x i32> %a, <4 x i32> %b, <4 x i32> %x, <4 x i32> %y) {368; CHECK-SD-LABEL: test_v4i32:369; CHECK-SD: // %bb.0:370; CHECK-SD-NEXT: sub v2.4s, v2.4s, v3.4s371; CHECK-SD-NEXT: cmhi v0.4s, v1.4s, v0.4s372; CHECK-SD-NEXT: add v0.4s, v2.4s, v0.4s373; CHECK-SD-NEXT: ret374;375; CHECK-GI-LABEL: test_v4i32:376; CHECK-GI: // %bb.0:377; CHECK-GI-NEXT: movi v4.4s, #1378; CHECK-GI-NEXT: cmhi v0.4s, v1.4s, v0.4s379; CHECK-GI-NEXT: sub v1.4s, v2.4s, v3.4s380; CHECK-GI-NEXT: and v0.16b, v0.16b, v4.16b381; CHECK-GI-NEXT: sub v0.4s, v1.4s, v0.4s382; CHECK-GI-NEXT: ret383 %cc = icmp ult <4 x i32> %a, %b384 %carry = zext <4 x i1> %cc to <4 x i32>385 %sub = sub <4 x i32> %x, %y386 %res = sub <4 x i32> %sub, %carry387 ret <4 x i32> %res388}389 390declare void @use()391;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:392; CHECK: {{.*}}393