non-trapping loads fix

1 files changed, 11 insertions, 3 deletions

diff --git a/scheduling/BTLScheduleraux.ml b/scheduling/BTLScheduleraux.ml
index c8c4e0d3..00be2aa7 100644
--- a/scheduling/BTLScheduleraux.ml
+++ b/scheduling/BTLScheduleraux.ml
@@ -120,9 +120,17 @@ let rec do_schedule btl = function
   | [] -> btl
   | (n, ibf) :: blks ->
       let code_exp = expanse n ibf btl in
-      let code_nt = turn_all_loads_nontrap n ibf code_exp in
-      let btl' = schedule_blk n ibf code_nt in
-      do_schedule btl' blks
+      let ibf_exp = get_some @@ PTree.get n code_exp in
+      let code_nt = turn_all_loads_nontrap n ibf_exp code_exp in
+      let ibf_nt = get_some @@ PTree.get n code_nt in
+      let btl' = schedule_blk n ibf_nt code_nt in
+      begin
+        (*debug_flag := true;*)
+        print_btl_code stderr code_nt;
+        print_btl_code stderr btl';
+        (*debug_flag := false;*)
+        do_schedule btl' blks
+      end
 
 let btl_scheduler f =
   let btl = f.fn_code in