Fixing bug with jlev loops and jlon loops nested inside.

ecmwf-ifs · Sep 12, 2024 · 47478f7 · 47478f7
1 parent a905b7a
commit 47478f7
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/transformations/transformations/parallel_routine_dispatch.py b/transformations/transformations/parallel_routine_dispatch.py
@@ -636,11 +636,14 @@ def process_loops(self, routine, region, map_routine, map_region, scc):
         else:
             lcpg_bnds =  map_routine['lcpg_bnds']
             for loop in FindNodes(ir.Loop).visit(region.body):
-                    lower_bound = routine.resolve_typebound_var(f"{lcpg_bnds}%KIDIA")
-                    upper_bound = routine.resolve_typebound_var(f"{lcpg_bnds}%KFDIA")
-                    new_bounds = sym.LoopRange((lower_bound, upper_bound))
-                    new_loop = loop.clone(bounds=new_bounds)
-                    loop_map[loop] = new_loop
+                    if loop.variable.name == "JLON":
+                        lower_bound = routine.resolve_typebound_var(f"{lcpg_bnds}%KIDIA")
+                        upper_bound = routine.resolve_typebound_var(f"{lcpg_bnds}%KFDIA")
+                        new_bounds = sym.LoopRange((lower_bound, upper_bound))
+                        new_loop = loop.clone(bounds=new_bounds)
+                        loop_map[loop] = new_loop
+                    #else:
+                    #    loop_map[loop] = loop
 #         new_region_body=Transformer(loop_map).visit(new_region_body)