234 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py2; RUN: llc --mtriple=loongarch32 -mattr=+d < %s | FileCheck %s --check-prefix=LA323; RUN: llc --mtriple=loongarch64 -mattr=+d < %s | FileCheck %s --check-prefix=LA644 5;; Compare if positive and select variable or zero.6define i8 @pos_sel_variable_and_zero_i8(i8 signext %a, i8 signext %b) {7; LA32-LABEL: pos_sel_variable_and_zero_i8:8; LA32: # %bb.0:9; LA32-NEXT: srai.w $a0, $a0, 710; LA32-NEXT: andn $a0, $a1, $a011; LA32-NEXT: ret12;13; LA64-LABEL: pos_sel_variable_and_zero_i8:14; LA64: # %bb.0:15; LA64-NEXT: srai.d $a0, $a0, 716; LA64-NEXT: andn $a0, $a1, $a017; LA64-NEXT: ret18 %cmp = icmp sgt i8 %a, -119 %sel = select i1 %cmp, i8 %b, i8 020 ret i8 %sel21}22 23define i16 @pos_sel_variable_and_zero_i16(i16 signext %a, i16 signext %b) {24; LA32-LABEL: pos_sel_variable_and_zero_i16:25; LA32: # %bb.0:26; LA32-NEXT: srai.w $a0, $a0, 1527; LA32-NEXT: andn $a0, $a1, $a028; LA32-NEXT: ret29;30; LA64-LABEL: pos_sel_variable_and_zero_i16:31; LA64: # %bb.0:32; LA64-NEXT: srai.d $a0, $a0, 1533; LA64-NEXT: andn $a0, $a1, $a034; LA64-NEXT: ret35 %cmp = icmp sgt i16 %a, -136 %sel = select i1 %cmp, i16 %b, i16 037 ret i16 %sel38}39 40define i32 @pos_sel_variable_and_zero_i32(i32 signext %a, i32 signext %b) {41; LA32-LABEL: pos_sel_variable_and_zero_i32:42; LA32: # %bb.0:43; LA32-NEXT: srai.w $a0, $a0, 3144; LA32-NEXT: andn $a0, $a1, $a045; LA32-NEXT: ret46;47; LA64-LABEL: pos_sel_variable_and_zero_i32:48; LA64: # %bb.0:49; LA64-NEXT: srai.d $a0, $a0, 3150; LA64-NEXT: andn $a0, $a1, $a051; LA64-NEXT: ret52 %cmp = icmp sgt i32 %a, -153 %sel = select i1 %cmp, i32 %b, i32 054 ret i32 %sel55}56 57define i64 @pos_sel_variable_and_zero_i64(i64 signext %a, i64 signext %b) {58; LA32-LABEL: pos_sel_variable_and_zero_i64:59; LA32: # %bb.0:60; LA32-NEXT: srai.w $a1, $a1, 3161; LA32-NEXT: andn $a0, $a2, $a162; LA32-NEXT: andn $a1, $a3, $a163; LA32-NEXT: ret64;65; LA64-LABEL: pos_sel_variable_and_zero_i64:66; LA64: # %bb.0:67; LA64-NEXT: srai.d $a0, $a0, 6368; LA64-NEXT: andn $a0, $a1, $a069; LA64-NEXT: ret70 %cmp = icmp sgt i64 %a, -171 %sel = select i1 %cmp, i64 %b, i64 072 ret i64 %sel73}74 75;; Compare if not negative or zero and select the same variable as being76;; compared: smax(a, 0).77define i8 @not_neg_not_zero_sel_same_variable_i8(i8 signext %a) {78; LA32-LABEL: not_neg_not_zero_sel_same_variable_i8:79; LA32: # %bb.0:80; LA32-NEXT: srai.w $a1, $a0, 781; LA32-NEXT: andn $a0, $a0, $a182; LA32-NEXT: ret83;84; LA64-LABEL: not_neg_not_zero_sel_same_variable_i8:85; LA64: # %bb.0:86; LA64-NEXT: srai.d $a1, $a0, 787; LA64-NEXT: andn $a0, $a0, $a188; LA64-NEXT: ret89 %cmp = icmp sgt i8 %a, 090 %sel = select i1 %cmp, i8 %a, i8 091 ret i8 %sel92}93 94define i16 @not_neg_not_zero_sel_same_variable_i16(i16 signext %a) {95; LA32-LABEL: not_neg_not_zero_sel_same_variable_i16:96; LA32: # %bb.0:97; LA32-NEXT: srai.w $a1, $a0, 1598; LA32-NEXT: andn $a0, $a0, $a199; LA32-NEXT: ret100;101; LA64-LABEL: not_neg_not_zero_sel_same_variable_i16:102; LA64: # %bb.0:103; LA64-NEXT: srai.d $a1, $a0, 15104; LA64-NEXT: andn $a0, $a0, $a1105; LA64-NEXT: ret106 %cmp = icmp sgt i16 %a, 0107 %sel = select i1 %cmp, i16 %a, i16 0108 ret i16 %sel109}110 111define i32 @not_neg_not_zero_sel_same_variable_i32(i32 signext %a) {112; LA32-LABEL: not_neg_not_zero_sel_same_variable_i32:113; LA32: # %bb.0:114; LA32-NEXT: srai.w $a1, $a0, 31115; LA32-NEXT: andn $a0, $a0, $a1116; LA32-NEXT: ret117;118; LA64-LABEL: not_neg_not_zero_sel_same_variable_i32:119; LA64: # %bb.0:120; LA64-NEXT: srai.d $a1, $a0, 31121; LA64-NEXT: andn $a0, $a0, $a1122; LA64-NEXT: ret123 %cmp = icmp sgt i32 %a, 0124 %sel = select i1 %cmp, i32 %a, i32 0125 ret i32 %sel126}127 128define i64 @not_neg_not_zero_sel_same_variable_i64(i64 signext %a) {129; LA32-LABEL: not_neg_not_zero_sel_same_variable_i64:130; LA32: # %bb.0:131; LA32-NEXT: srai.w $a2, $a1, 31132; LA32-NEXT: andn $a0, $a0, $a2133; LA32-NEXT: andn $a1, $a1, $a2134; LA32-NEXT: ret135;136; LA64-LABEL: not_neg_not_zero_sel_same_variable_i64:137; LA64: # %bb.0:138; LA64-NEXT: srai.d $a1, $a0, 63139; LA64-NEXT: andn $a0, $a0, $a1140; LA64-NEXT: ret141 %cmp = icmp sgt i64 %a, 0142 %sel = select i1 %cmp, i64 %a, i64 0143 ret i64 %sel144}145 146;; ret = (x-y) > 0 ? x-y : 0147define i8 @sub_clamp_zero_i8(i8 signext %x, i8 signext %y) {148; LA32-LABEL: sub_clamp_zero_i8:149; LA32: # %bb.0:150; LA32-NEXT: sub.w $a0, $a0, $a1151; LA32-NEXT: slli.w $a1, $a0, 24152; LA32-NEXT: srai.w $a1, $a1, 31153; LA32-NEXT: andn $a0, $a0, $a1154; LA32-NEXT: ret155;156; LA64-LABEL: sub_clamp_zero_i8:157; LA64: # %bb.0:158; LA64-NEXT: sub.d $a0, $a0, $a1159; LA64-NEXT: ext.w.b $a1, $a0160; LA64-NEXT: srai.d $a1, $a1, 7161; LA64-NEXT: andn $a0, $a0, $a1162; LA64-NEXT: ret163 %sub = sub nsw i8 %x, %y164 %cmp = icmp sgt i8 %sub, 0165 %sel = select i1 %cmp, i8 %sub, i8 0166 ret i8 %sel167}168 169define i16 @sub_clamp_zero_i16(i16 signext %x, i16 signext %y) {170; LA32-LABEL: sub_clamp_zero_i16:171; LA32: # %bb.0:172; LA32-NEXT: sub.w $a0, $a0, $a1173; LA32-NEXT: slli.w $a1, $a0, 16174; LA32-NEXT: srai.w $a1, $a1, 31175; LA32-NEXT: andn $a0, $a0, $a1176; LA32-NEXT: ret177;178; LA64-LABEL: sub_clamp_zero_i16:179; LA64: # %bb.0:180; LA64-NEXT: sub.d $a0, $a0, $a1181; LA64-NEXT: ext.w.h $a1, $a0182; LA64-NEXT: srai.d $a1, $a1, 15183; LA64-NEXT: andn $a0, $a0, $a1184; LA64-NEXT: ret185 %sub = sub nsw i16 %x, %y186 %cmp = icmp sgt i16 %sub, 0187 %sel = select i1 %cmp, i16 %sub, i16 0188 ret i16 %sel189}190 191define i32 @sub_clamp_zero_i32(i32 signext %x, i32 signext %y) {192; LA32-LABEL: sub_clamp_zero_i32:193; LA32: # %bb.0:194; LA32-NEXT: sub.w $a0, $a0, $a1195; LA32-NEXT: srai.w $a1, $a0, 31196; LA32-NEXT: andn $a0, $a0, $a1197; LA32-NEXT: ret198;199; LA64-LABEL: sub_clamp_zero_i32:200; LA64: # %bb.0:201; LA64-NEXT: sub.w $a0, $a0, $a1202; LA64-NEXT: srai.d $a1, $a0, 31203; LA64-NEXT: andn $a0, $a0, $a1204; LA64-NEXT: ret205 %sub = sub nsw i32 %x, %y206 %cmp = icmp sgt i32 %sub, 0207 %sel = select i1 %cmp, i32 %sub, i32 0208 ret i32 %sel209}210 211define i64 @sub_clamp_zero_i64(i64 signext %x, i64 signext %y) {212; LA32-LABEL: sub_clamp_zero_i64:213; LA32: # %bb.0:214; LA32-NEXT: sltu $a4, $a0, $a2215; LA32-NEXT: sub.w $a1, $a1, $a3216; LA32-NEXT: sub.w $a1, $a1, $a4217; LA32-NEXT: sub.w $a0, $a0, $a2218; LA32-NEXT: srai.w $a2, $a1, 31219; LA32-NEXT: andn $a1, $a1, $a2220; LA32-NEXT: andn $a0, $a0, $a2221; LA32-NEXT: ret222;223; LA64-LABEL: sub_clamp_zero_i64:224; LA64: # %bb.0:225; LA64-NEXT: sub.d $a0, $a0, $a1226; LA64-NEXT: srai.d $a1, $a0, 63227; LA64-NEXT: andn $a0, $a0, $a1228; LA64-NEXT: ret229 %sub = sub nsw i64 %x, %y230 %cmp = icmp sgt i64 %sub, 0231 %sel = select i1 %cmp, i64 %sub, i64 0232 ret i64 %sel233}234