Changeset 744 for codes/icosagcm

Timestamp:

10/03/18 17:58:55 (6 years ago)

Author:

dubos

Message:

devel : make_icosa option for mixed precision

Location:

codes/icosagcm/devel

Files:

• Python/src/setup.py (modified) (1 diff)
• make_icosa (modified) (3 diffs)
• src/unstructured/data_unstructured.F90 (modified) (1 diff)
• src/unstructured/unstructured.h90 (modified) (1 diff)

codes/icosagcm/devel/Python/src/setup.py

@@ -2 +2 @@
 from distutils.extension import Extension
 from Cython.Build import cythonize
+
+from ctypes import cdll, c_bool
+libicosa = cdll.LoadLibrary('libicosa.so')
+mixed_prec=c_bool.in_dll(libicosa, 'mixed_precision').value
+
+if mixed_prec:
+    extra_flags=["-DCPP_MIXED_PREC"]
+else:
+    extra_flags=None
+
 setup(  name = "DYNAMICO kernels for unstructured mesh",
         ext_modules = cythonize([
             Extension("cxios",["cxios.pyx"],libraries=["xios"]),
-            Extension("unstructured",["unstructured.pyx"],libraries=["icosa","xios"])
-            ])
+            Extension("unstructured",["unstructured.pyx"],libraries=["icosa","xios"], extra_compile_args=extra_flags)
+            ], compile_time_env={'CPP_MIXED_PREC': mixed_prec} )
         )

codes/icosagcm/devel/make_icosa

@@ -44 +44 @@
  -physics lmdz_generic     : use LMDZ generic physics (planets)
  -external_physics         : allows external physics
+ -mixed_prec               : compiles unstructured dynamics in single precision (float32)
 fin
           exit;;
@@ -86 +87 @@
           external_physics="TRUE" ; shift ;;
 
+      "-mixed_prec")
+          mixed_prec="TRUE" ; shift ;;
       *)
           code="$1" ; shift ;;
@@ -183 +186 @@
 fi
 
+if [[ "$mixed_prec" == "TRUE" ]]
+then
+  CPP_KEY="$CPP_KEY CPP_MIXED_PREC"
+fi
+
 if [[ "$no_io_defined" == "FALSE" ]]
 then 

codes/icosagcm/devel/src/unstructured/data_unstructured.F90

@@ -13 +13 @@
        caldyn_vert_variant=caldyn_vert_cons, nb_threads=0, nb_stage=0
   LOGICAL(C_BOOL), BIND(C) :: hydrostatic=.TRUE., debug_hevi_solver=.TRUE.
+
+#ifdef CPP_MIXED_PREC
+  LOGICAL(C_BOOL), BIND(C) :: mixed_precision=.TRUE.
+#else
+  LOGICAL(C_BOOL), BIND(C) :: mixed_precision=.FALSE.
+#endif
 
   INDEX, BIND(C) :: llm, nqdyn, edge_num, primal_num, dual_num, &

codes/icosagcm/devel/src/unstructured/unstructured.h90

@@ -1 @@
-#undef PREC_DOUBLE
-
 #define DBL REAL(C_DOUBLE)
 #define FLT REAL(C_FLOAT)
 #define TIME DBL
 
-#ifdef PREC_DOUBLE
+#ifdef CPP_MIXED_PREC
+#define NUM FLT
+#define MPI_NUM MPI_REAL4
+#else
 #define NUM DBL
 #define MPI_NUM MPI_REAL8
-#else
-#define NUM FLT
-#define MPI_NUM MPI_REAL4
 #endif