[llvm] [InferAS] Support getAssumedAddrSpace for Arguments for NVPTX (PR #133991)

[Alex MacLean via llvm-commits]

================
@@ -1275,7 +1294,7 @@ void InferAddressSpacesImpl::performPointerReplacement(
     // This instruction may contain multiple uses of V, update them all.
     CurUser->replaceUsesOfWith(
         V, new AddrSpaceCastInst(NewV, V->getType(), "", InsertPos));
-  } else {
+  } else if (isa<Constant>(V)) {
     CurUserI->replaceUsesOfWith(
         V, ConstantExpr::getAddrSpaceCast(cast<Constant>(NewV), V->getType()));
   }
----------------
AlexMaclean wrote:

Good point, I've switched this back to catch-all handling and updated the above `if` branch to handle arguments. I'm surprising the previous version seemed to work at all 😅 

https://github.com/llvm/llvm-project/pull/133991