fix getSource for splat

Author: sun-jacobi

llvm/lib/Target/RISCV/RISCVISelLowering.cpp

@@ -13598,17 +13598,6 @@ struct NodeExtensionHelper {
     case RISCVISD::VZEXT_VL:
     case RISCVISD::FP_EXTEND_VL:
       return OrigOperand.getOperand(0);
-    case ISD::SPLAT_VECTOR: {
-      SDValue Op = OrigOperand.getOperand(0);
-      unsigned Opc = Op.getOpcode();
-      if (SupportsSExt && Opc == ISD::SIGN_EXTEND_INREG)
-        return Op.getOperand(0);
-
-      if (SupportsZExt && Opc == ISD::AND)
-        return Op.getOperand(0);
-
-      return Op;
-    }
     default:
       return OrigOperand;
     }
@@ -13662,7 +13651,7 @@ struct NodeExtensionHelper {
     case RISCVISD::FP_EXTEND_VL:
       return DAG.getNode(ExtOpc, DL, NarrowVT, Source, Mask, VL);
     case ISD::SPLAT_VECTOR:
-      return DAG.getSplat(NarrowVT, DL, Source);
+      return DAG.getSplat(NarrowVT, DL, Source.getOperand(0));
     case RISCVISD::VMV_V_X_VL:
       return DAG.getNode(RISCVISD::VMV_V_X_VL, DL, NarrowVT,
                          DAG.getUNDEF(NarrowVT), Source.getOperand(1), VL);

llvm/test/CodeGen/RISCV/rvv/vwadd-sdnode.ll

@@ -1484,6 +1484,7 @@ define <vscale x 8 x i64> @vwadd_vx_splat_zext(<vscale x 8 x i32> %va, i32 %b) {
 ;
 ; RV64-LABEL: vwadd_vx_splat_zext:
 ; RV64:       # %bb.0:
+; RV64-NEXT:    andi a0, a0, -1
 ; RV64-NEXT:    vsetvli a1, zero, e32, m4, ta, ma
 ; RV64-NEXT:    vwaddu.vx v16, v8, a0
 ; RV64-NEXT:    vmv8r.v v8, v16