Changeset 721 for codes/icosagcm/devel/Python

Timestamp:

08/08/18 19:37:11 (6 years ago)

Author:

dubos

Message:

devel : created serial kernels to debug OpenMP

Location:

codes/icosagcm/devel/Python/src

Files:

• hex_master (added)
• hex_master/macros.jin (added)
• hexagonal/macros.jin (modified) (6 diffs)

codes/icosagcm/devel/Python/src/hexagonal/macros.jin

@@ -58 +58 @@
 #if PASS_JINJA
 
+#ifdef HEX_MASTER
+
+{% set ll_begin, ll_beginp1 = '1', '2' %}  
+{% set ll_endm1, ll_end, ll_endp1 = 'llm-1', 'llm', 'llm+1' %}  
+{% macro ij_omp(ext) -%}
+ij_begin{{ext}},ij_end{{ext}}
+{%- endmacro %}
+
+#else
+
+{% set ll_begin, ll_beginp1 = 'll_begin', 'll_beginp1' %}  
+{% set ll_endm1, ll_end, ll_endp1 = 'll_endm1', 'll_end', 'll_endp1' %}  
+{% macro ij_omp(ext) -%}
+ij_omp_begin{{ext}},ij_omp_end{{ext}}
+{%- endmacro %}
+
+#endif
+
 {% macro sequence() -%}
 {{ caller(body_primal,at_level_primal,'') }}
@@ -68 +86 @@
 {% macro at_level_primal(lev,ext) -%}
 {{ define('LL', lev) }}
-DO ij=ij_omp_begin{{ext}},ij_omp_end{{ext}}
+DO ij={{ ij_omp(ext) }}
 {{ caller() }}
 END DO
@@ -76 +94 @@
 {{ define('LL', 'l') }}
 DO l = {{ range }}
-  DO ij=ij_omp_begin{{ext}},ij_omp_end{{ext}}
+  DO ij={{ ij_omp(ext) }}
     {{ caller() }}
   END DO
@@ -88 +106 @@
 {% macro at_level_edge(lev,ext) -%}
 {{ define('LL', lev) }}
-DO ij=ij_omp_begin{{ext}},ij_omp_end{{ext}}
+DO ij={{ ij_omp(ext) }}
       {{ caller('u_right','t_right') }}
       {{ caller('u_lup','t_lup') }}
@@ -98 +116 @@
 {{ define('LL', 'l') }}
 DO l = {{ range }}
-   DO ij=ij_omp_begin{{ext}},ij_omp_end{{ext}}
+   DO ij={{ ij_omp(ext) }}
       {{ caller('u_right','t_right') }}
       {{ caller('u_lup','t_lup') }}
@@ -117 +135 @@
 
 {% macro forall(start,end,ij_range,thecode) %}
-{% set start_omp={'1':'ll_begin' , '2':'ll_beginp1'}[start] %}  
-{% set end_omp={'llm-1':'ll_endm1' , 'llm':'ll_end', 'llm+1':'ll_endp1'}[end] %}
+{% set start_omp={'1':ll_begin , '2':ll_beginp1}[start] %}  
+{% set end_omp={'llm-1':ll_endm1 , 'llm':ll_end, 'llm+1':ll_endp1}[end] %}
 {% set is_top_layer, is_top_inter = False, False %}