add extra judgement

tangaac · tangaac · commit e412c722f7b3 · 2025-04-16T11:51:32.000+08:00
diff --git a/llvm/lib/Target/LoongArch/LoongArchISelLowering.cpp b/llvm/lib/Target/LoongArch/LoongArchISelLowering.cpp
@@ -1748,6 +1748,9 @@ static SDValue lowerBUILD_VECTORAsBroadCastLoad(BuildVectorSDNode *BVOp,
     IdentitySrc = BVOp->getOperand(0);
   }
 
+  if (!BVOp->isOnlyUserOf(IdentitySrc.getNode()))
+    return SDValue();
+
   if (IsIdeneity) {
     auto *LN = cast<LoadSDNode>(IdentitySrc);
     SDVTList Tys =

Original file line number	Diff line number	Diff line change
`@@ -1748,6 +1748,9 @@ static SDValue lowerBUILD_VECTORAsBroadCastLoad(BuildVectorSDNode *BVOp,`
`1748`	`1748`	`IdentitySrc = BVOp->getOperand(0);`
`1749`	`1749`	`}`
`1750`	`1750`
	`1751`	`+ if (!BVOp->isOnlyUserOf(IdentitySrc.getNode()))`
	`1752`	`+ return SDValue();`
	`1753`	`+`
`1751`	`1754`	`if (IsIdeneity) {`
`1752`	`1755`	`auto *LN = cast<LoadSDNode>(IdentitySrc);`
`1753`	`1756`	`SDVTList Tys =`