Call findAssignmentConstDiff on loops body instead the loop itself

goblint · Sep 28, 2024 · da52931 · da52931
1 parent e77b41c
commit da52931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/util/loopUnrolling.ml b/src/util/loopUnrolling.ml
@@ -262,7 +262,7 @@ let fixedLoopSize loopStatement func =
   in let assignmentDifference loop var = try
          let diff = ref None in
          let visitor = new findAssignmentConstDiff(diff, var) in
-         ignore @@ visitCilStmt visitor loop;
+         ignore @@ visitCilBlock visitor loop;
          !diff
        with | WrongOrMultiple ->  None
   in
@@ -273,7 +273,7 @@ let fixedLoopSize loopStatement func =
     None
   else
     constBefore var loopStatement func >>= fun start ->
-    assignmentDifference loopStatement var >>= fun diff ->
+    assignmentDifference (loopBody loopStatement) var >>= fun diff ->
     Logs.debug "comparison: ";
     Pretty.fprint stderr (dn_exp () comparison) ~width:max_int;
     Logs.debug "";