Adding pathmap psize modification during expansion oracle

1 files changed, 5 insertions, 5 deletions

diff --git a/scheduling/RTLpathScheduleraux.ml b/scheduling/RTLpathScheduleraux.ml
index cdee5a5b..19b05741 100644
--- a/scheduling/RTLpathScheduleraux.ml
+++ b/scheduling/RTLpathScheduleraux.ml
@@ -280,11 +280,11 @@ let turn_all_loads_nontrap sb code =
     !code'
   end
 
-let rec do_schedule code = function
-  | [] -> code
+let rec do_schedule code pm = function
+  | [] -> (code, pm)
   | sb :: lsb ->
       (*debug_flag := true;*)
-      let code_exp = expanse sb code in
+      let (code_exp, pm) = expanse sb code pm in
       (*debug_flag := false;*)
       (* Trick: instead of turning loads into non trap as needed..
        * First, we turn them all into non-trap.
@@ -302,7 +302,7 @@ let rec do_schedule code = function
         debug "\nNew Code: "; print_code new_code;
         debug "\n";
         (* debug_flag := false; *)
-        do_schedule new_code lsb
+        do_schedule new_code pm lsb
       end
 
 let get_ok r = match r with Errors.OK x -> x | _ -> failwith "Did not get OK"
@@ -323,6 +323,6 @@ let scheduler f =
     (*debug_flag := true; *)
     find_last_node_reg (PTree.elements code);
     (*print_code code;*)
-    let tc = do_schedule code lsb in
+    let (tc, pm) = do_schedule code pm lsb in
     (((tc, entry), pm), id_ptree)
   end