Cast of addr13 (#2147)

EnzymeAD · Nov 3, 2024 · 1bcf5ee · 1bcf5ee
1 parent b6ab325
commit 1bcf5ee
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/enzyme/Enzyme/DifferentialUseAnalysis.cpp b/enzyme/Enzyme/DifferentialUseAnalysis.cpp
@@ -917,6 +917,15 @@ void DifferentialUseAnalysis::minCut(const DataLayout &DL, LoopInfo &OrigLI,
         if (ASC->getSrcAddressSpace() == 10 && ASC->getDestAddressSpace() == 0)
           continue;
       }
+      if (auto CI = dyn_cast<CastInst>((*mp.begin()).V)) {
+        if (CI->getType()->isPointerTy() &&
+            CI->getType()->getPointerAddressSpace() == 13)
+          continue;
+      }
+      if (auto G = dyn_cast<GetElementPtrInst>((*mp.begin()).V)) {
+        if (G->getType()->getPointerAddressSpace() == 13)
+          continue;
+      }
       if (hasNoCache((*mp.begin()).V)) {
         continue;
       }