All loaded values extension

enzyme/Enzyme/ActivityAnalysis.cpp

@@ -2697,8 +2697,8 @@ bool ActivityAnalyzer::isValueInactiveFromUsers(TypeResults const &TR,
     }
 
     if (EnzymePrintActivity)
-      llvm::errs() << "      considering use of " << *val << " - " << *a
-                   << "\n";
+      llvm::errs() << "      considering use(" << to_string(UA) << ") of "
+                   << *val << " - " << *a << "\n";
 
     // Only ignore stores to the operand, not storing the operand
     // somewhere

enzyme/Enzyme/Utils.cpp

@@ -2627,7 +2627,7 @@ bool writesToMemoryReadBy(const TypeResults *TR, llvm::AAResults &AA,
 
 // Find the base pointer of ptr and the offset in bytes from the start of
 // the returned base pointer to this value.
-AllocaInst *getBaseAndOffset(Value *ptr, size_t &offset) {
+Value *getBaseAndOffset(Value *ptr, size_t &offset) {
   offset = 0;
   while (true) {
     if (auto CI = dyn_cast<CastInst>(ptr)) {
@@ -2663,7 +2663,16 @@ AllocaInst *getBaseAndOffset(Value *ptr, size_t &offset) {
     }
     return nullptr;
   }
-  return cast<AllocaInst>(ptr);
+  if (isa<AllocaInst>(ptr))
+    return ptr;
+  // Assume all allocainsts are uninitialized data
+  if (auto *CI = llvm::dyn_cast<llvm::CallBase>(ptr)) {
+    auto name = getFuncNameFromCall(CI);
+    if (name == "malloc" || name == "julia.gc_alloc_obj") {
+      return ptr;
+    }
+  }
+  return nullptr;
 }
 
 // Find all user instructions of AI, returning tuples of <instruction, value,
@@ -2715,8 +2724,7 @@ findAllUsersOf(Value *AI) {
 // that pointer when indexed at offset. If it is impossible to guarantee that
 // the set contains all such values, set legal to false
 SmallVector<std::pair<Value *, size_t>, 1>
-getAllLoadedValuesFrom(AllocaInst *ptr0, size_t offset, size_t valSz,
-                       bool &legal) {
+getAllLoadedValuesFrom(Value *ptr0, size_t offset, size_t valSz, bool &legal) {
   SmallVector<std::pair<Value *, size_t>, 1> options;
 
   auto todo = findAllUsersOf(ptr0);

enzyme/Enzyme/Utils.h

@@ -2121,4 +2121,8 @@ arePointersGuaranteedNoAlias(llvm::TargetLibraryInfo &TLI, llvm::AAResults &AA,
                              llvm::LoopInfo &LI, llvm::Value *op0,
                              llvm::Value *op1, bool offsetAllowed = false);
 
+llvm::SmallVector<std::pair<llvm::Value *, size_t>, 1>
+getAllLoadedValuesFrom(llvm::Value *ptr0, size_t offset, size_t valSz,
+                       bool &legal);
+
 #endif // ENZYME_UTILS_H