Changeset 744 for codes/icosagcm/devel/Python/src/setup.py

Timestamp:

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

Author:

dubos

Message:

devel : make_icosa option for mixed precision

File:

• codes/icosagcm/devel/Python/src/setup.py (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} )
         )