Address comments

Author: aengelke

llvm/lib/Analysis/ValueTracking.cpp

@@ -526,10 +526,6 @@ bool llvm::isValidAssumeForContext(const Instruction *Inv,
     return AllowEphemerals || !isEphemeralValueOf(Inv, CxtI);
   }
 
-  // Inv and CxtI are in different functions.
-  if (Inv->getFunction() != CxtI->getFunction())
-    return false;
-
   // Inv and CxtI are in different blocks.
   if (DT) {
     if (DT->dominates(Inv, CxtI))

llvm/lib/Transforms/Scalar/AlignmentFromAssumptions.cpp

@@ -208,6 +208,7 @@ bool AlignmentFromAssumptionsPass::processAssumption(CallInst *ACall,
       continue;
 
     if (Instruction *K = dyn_cast<Instruction>(J))
+      if (K->getFunction() == ACall->getFunction())
         WorkList.push_back(K);
   }
 
@@ -267,6 +268,8 @@ bool AlignmentFromAssumptionsPass::processAssumption(CallInst *ACall,
       for (auto &U : J->uses()) {
         if (U->getType()->isPointerTy()) {
           Instruction *K = cast<Instruction>(U.getUser());
+          if (K->getFunction() != ACall->getFunction())
+            continue;
           StoreInst *SI = dyn_cast<StoreInst>(K);
           if (SI && SI->getPointerOperandIndex() != U.getOperandNo())
             continue;