Set _threading_type at compile time

This variable can be used at run-time to detect whether compilation was done against OpenMP, PTHREADS or without multithreading.
s-sajid-ali · Sep 17, 2018 · e17d3c6 · e17d3c6
1 parent 2f9d3c0
commit e17d3c6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyfftw/pyfftw.pyx b/pyfftw/pyfftw.pyx
@@ -80,6 +80,13 @@ IF HAVE_LONG:
     _supported_nptypes_complex.append(np.clongdouble)
     _supported_nptypes_real.append(np.longdouble)
 
+IF (HAVE_SINGLE_OMP or HAVE_DOUBLE_OMP or HAVE_LONG_OMP):
+    _threading_type = 'OMP'
+ELIF (HAVE_SINGLE_THREADS or HAVE_DOUBLE_THREADS or HAVE_LONG_THREADS):
+    _threading_type = 'PTHREADS'
+ELSE:
+    _threading_type = None
+
 cdef object directions
 directions = {'FFTW_FORWARD': FFTW_FORWARD,
         'FFTW_BACKWARD': FFTW_BACKWARD}