diff --git a/llvm/test/CodeGen/RISCV/rvv/vwadd-sdnode.ll b/llvm/test/CodeGen/RISCV/rvv/vwadd-sdnode.ll index 36bc10f055b84..ed12afdd95956 100644 --- a/llvm/test/CodeGen/RISCV/rvv/vwadd-sdnode.ll +++ b/llvm/test/CodeGen/RISCV/rvv/vwadd-sdnode.ll @@ -1396,8 +1396,8 @@ define @i1_zext( %va, %vb ; %x.i32 and %y.i32 are disjoint, so DAGCombiner will combine it into an or. ; FIXME: We should be able to recover the or into vwaddu.vv if the disjoint ; flag is set. -define @disjoint_or( %x.i8, %y.i8) { -; CHECK-LABEL: disjoint_or: +define @vwaddu_vv_disjoint_or_add( %x.i8, %y.i8) { +; CHECK-LABEL: vwaddu_vv_disjoint_or_add: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16, mf2, ta, ma ; CHECK-NEXT: vzext.vf2 v10, v8 @@ -1414,3 +1414,59 @@ define @disjoint_or( %x.i8, %x.i32, %y.i32 ret %add } + +; TODO: We could select vwaddu.vv, but when both arms of the or are the same +; DAGCombiner::hoistLogicOpWithSameOpcodeHands moves the zext above the or. +define @vwaddu_vv_disjoint_or( %x.i16, %y.i16) { +; CHECK-LABEL: vwaddu_vv_disjoint_or: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e16, mf2, ta, ma +; CHECK-NEXT: vor.vv v9, v8, v9 +; CHECK-NEXT: vsetvli zero, zero, e32, m1, ta, ma +; CHECK-NEXT: vzext.vf2 v8, v9 +; CHECK-NEXT: ret + %x.i32 = zext %x.i16 to + %y.i32 = zext %y.i16 to + %or = or disjoint %x.i32, %y.i32 + ret %or +} + +; TODO: We could select vwadd.vv, but when both arms of the or are the same +; DAGCombiner::hoistLogicOpWithSameOpcodeHands moves the zext above the or. +define @vwadd_vv_disjoint_or( %x.i16, %y.i16) { +; CHECK-LABEL: vwadd_vv_disjoint_or: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e16, mf2, ta, ma +; CHECK-NEXT: vor.vv v9, v8, v9 +; CHECK-NEXT: vsetvli zero, zero, e32, m1, ta, ma +; CHECK-NEXT: vsext.vf2 v8, v9 +; CHECK-NEXT: ret + %x.i32 = sext %x.i16 to + %y.i32 = sext %y.i16 to + %or = or disjoint %x.i32, %y.i32 + ret %or +} + +define @vwaddu_wv_disjoint_or( %x.i32, %y.i16) { +; CHECK-LABEL: vwaddu_wv_disjoint_or: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, ma +; CHECK-NEXT: vzext.vf2 v10, v9 +; CHECK-NEXT: vor.vv v8, v8, v10 +; CHECK-NEXT: ret + %y.i32 = zext %y.i16 to + %or = or disjoint %x.i32, %y.i32 + ret %or +} + +define @vwadd_wv_disjoint_or( %x.i32, %y.i16) { +; CHECK-LABEL: vwadd_wv_disjoint_or: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, ma +; CHECK-NEXT: vsext.vf2 v10, v9 +; CHECK-NEXT: vor.vv v8, v8, v10 +; CHECK-NEXT: ret + %y.i32 = sext %y.i16 to + %or = or disjoint %x.i32, %y.i32 + ret %or +}