[llvm] [NVPTX] Add some basic folds for ADDRSPACECAST (PR #129157)

[Artem Belevich via llvm-commits]

================
@@ -5209,6 +5209,26 @@ PerformBUILD_VECTORCombine(SDNode *N, TargetLowering::DAGCombinerInfo &DCI) {
   return DAG.getNode(ISD::BITCAST, DL, VT, PRMT);
 }
 
+static SDValue combineADDRSPACECAST(SDNode *N,
+                                    TargetLowering::DAGCombinerInfo &DCI) {
+  auto *ASCN1 = cast<AddrSpaceCastSDNode>(N);
+
+  if (auto *ASCN2 = dyn_cast<AddrSpaceCastSDNode>(ASCN1->getOperand(0))) {
+    assert(ASCN2->getDestAddressSpace() == ASCN1->getSrcAddressSpace());
+
+    // Fold asc[B -> A](asc[A -> B](x)) -> x
+    if (ASCN1->getDestAddressSpace() == ASCN2->getSrcAddressSpace())
+      return ASCN2->getOperand(0);
----------------
Artem-B wrote:

I think we intentionally insert `ASC(generic->global->generic)` to allow tracking specific AS the pointer actually points to, so we can lower loads/stores using AS-specific instruction. Collapsing it all back to `generic` will lose the hint, if it kicks in before we've lowered loads/stores.


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