Special case gep of small_typeof (#1433)

EnzymeAD · Sep 18, 2023 · a7302ce · a7302ce
1 parent 4c6c0a6
commit a7302ce
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/enzyme/Enzyme/TypeAnalysis/TypeAnalysis.cpp b/enzyme/Enzyme/TypeAnalysis/TypeAnalysis.cpp
@@ -1401,6 +1401,16 @@ void TypeAnalyzer::visitGetElementPtrInst(GetElementPtrInst &gep) {
       return;
     }
   }
+  if (auto GV = dyn_cast<GlobalVariable>(gep.getPointerOperand())) {
+    // from julia code, do not propagate int to operands
+    if (GV->getName() == "small_typeof") {
+      TypeTree T;
+      T.insert({-1}, BaseType::Pointer);
+      T.insert({-1, -1}, BaseType::Pointer);
+      updateAnalysis(&gep, T, &gep);
+      return;
+    }
+  }
 
   if (gep.indices().begin() == gep.indices().end()) {
     if (direction & DOWN)