Fix load is load error (#1862)

EnzymeAD · May 7, 2024 · 7d09d5e · 7d09d5e
1 parent cbccccf
commit 7d09d5e
Show file tree

Hide file tree

Showing 2 changed files with 1,814 additions and 4 deletions.
diff --git a/enzyme/Enzyme/GradientUtils.cpp b/enzyme/Enzyme/GradientUtils.cpp
@@ -834,10 +834,7 @@ Value *GradientUtils::unwrapM(Value *const val, IRBuilder<> &BuilderM,
           if (orig && knownRecomputeHeuristic.find(orig) !=
                           knownRecomputeHeuristic.end()) {
             if (!knownRecomputeHeuristic[orig]) {
-              if (!legalRecompute(orig, available, &BuilderM))
-                return nullptr;
-
-              assert(isa<LoadInst>(orig) == isa<LoadInst>(val));
+              return nullptr;
             }
           }
         }