Fix integration tests on main

EnzymeAD · Feb 5, 2024 · d70efae · d70efae
1 parent 9ff56bc
commit d70efae
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/enzyme/Enzyme/AdjointGenerator.h b/enzyme/Enzyme/AdjointGenerator.h
@@ -3353,6 +3353,25 @@ class AdjointGenerator
             }
           }
         }
+        if (size == 1 && !isa<ConstantInt>(new_size)) {
+          for (auto ptr : {orig_dst, orig_src}) {
+            vd = TR.query(ptr).Data0().ShiftIndices(DL, 0, -1, 0);
+            if (vd.isKnownPastPointer()) {
+              ConcreteType mv(BaseType::Unknown);
+              size_t minInt = 0xFFFFFFFF;
+              for (const auto &pair : vd.getMapping()) {
+                if (pair.first.size() != 1)
+                  continue;
+                if (minInt < pair.first[0])
+                  continue;
+                minInt = pair.first[0];
+                mv = pair.second;
+              }
+              vd.insert({(int)minInt}, mv);
+              goto known;
+            }
+          }
+        }
         if (errorIfNoType)
           EmitWarning("CannotDeduceType", MTI, "failed to deduce type of copy ",
                       MTI);
@@ -3368,6 +3387,7 @@ class AdjointGenerator
                              &TR.analyzer, nullptr, wrap(&BuilderZ));
         } else {
           ss << "\n";
+          ss << *gutils->oldFunc << "\n";
           TR.dump(ss);
           EmitFailure("CannotDeduceType", MTI.getDebugLoc(), &MTI, ss.str());
         }

diff --git a/enzyme/test/Integration/ReverseMode/forrealloc.c b/enzyme/test/Integration/ReverseMode/forrealloc.c
@@ -1,6 +1,6 @@
 // RUN: %clang -std=c11 -O0 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli - 
 // RUN: %clang -std=c11 -O1 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli - 
-// RUN: %clang -std=c11 -O2 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli - 
+// RUN: %clang -std=c11 -O2 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S -enzyme-loose-types | %lli - 
 // RUN: %clang -std=c11 -O3 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -S | %lli - 
 // RUN: %clang -std=c11 -O0 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -enzyme-inline=1 -S | %lli - 
 // RUN: %clang -std=c11 -O1 %s -S -emit-llvm -o - | %opt - %OPloadEnzyme %enzyme -enzyme-inline=1 -S | %lli -