Add address space check

abhishek-kaushik22 · abhishek-kaushik22 · commit f4157dd778a6 · 2025-08-04T14:55:04.000+05:30
diff --git a/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp b/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
@@ -22569,7 +22569,8 @@ static SDValue foldToMaskedStore(StoreSDNode *Store, SelectionDAG &DAG,
   }
 
   auto *Load = cast<LoadSDNode>(StoredVal.getOperand(LoadPos));
-  if (!Load->isSimple() || !ISD::isNormalLoad(Load))
+  if (!Load->isSimple() || !ISD::isNormalLoad(Load) ||
+      Load->getAddressSpace() != AddrSpace)
     return SDValue();
 
   if (!Store->getChain().reachesChainWithoutSideEffects(LoadCh))

Original file line number	Diff line number	Diff line change
`@@ -22569,7 +22569,8 @@ static SDValue foldToMaskedStore(StoreSDNode *Store, SelectionDAG &DAG,`
`22569`	`22569`	`}`
`22570`	`22570`
`22571`	`22571`	`auto *Load = cast<LoadSDNode>(StoredVal.getOperand(LoadPos));`
`22572`		`- if (!Load->isSimple() \|\| !ISD::isNormalLoad(Load))`
	`22572`	`+ if (!Load->isSimple() \|\| !ISD::isNormalLoad(Load) \|\|`
	`22573`	`+ Load->getAddressSpace() != AddrSpace)`
`22573`	`22574`	`return SDValue();`
`22574`	`22575`
`22575`	`22576`	`if (!Store->getChain().reachesChainWithoutSideEffects(LoadCh))`