add nogil to MKL functions for freeing buffers

Author: ndgrigorian

mkl/_mkl_service.pxd

@@ -29,7 +29,7 @@ ctypedef unsigned long long MKL_UINT64
 ctypedef int MKL_INT
 
 
-cdef extern from "mkl.h":
+cdef extern from "mkl.h" nogil:
     # MKL Function Domains Constants
     int MKL_DOMAIN_BLAS
     int MKL_DOMAIN_FFT
@@ -151,10 +151,10 @@ cdef extern from "mkl.h":
     MKL_INT64 mkl_mem_stat(int* buf)
     MKL_INT64 mkl_peak_mem_usage(int mode)
     int mkl_set_memory_limit(int mem_type, size_t limit)
-    void *mkl_malloc(size_t size, int alignment) nogil
-    void *mkl_realloc(void *ptr, size_t size) nogil
-    void *mkl_calloc(size_t num, size_t size, int alignment) nogil
-    void mkl_free(void *ptr) nogil
+    void *mkl_malloc(size_t size, int alignment)
+    void *mkl_realloc(void *ptr, size_t size)
+    void *mkl_calloc(size_t num, size_t size, int alignment)
+    void mkl_free(void *ptr)
 
     # Conditional Numerical Reproducibility
     int mkl_cbwr_set(int settings)

mkl/_py_mkl_service.pyx

@@ -601,7 +601,8 @@ cdef inline void __free_buffers() noexcept:
     """
     Frees unused memory allocated by the Intel(R) MKL Memory Allocator.
     """
-    mkl.mkl_free_buffers()
+    with nogil:
+        mkl.mkl_free_buffers()
     return
 
 
@@ -610,7 +611,8 @@ cdef inline void __thread_free_buffers() noexcept:
     Frees unused memory allocated by the Intel(R) MKL Memory Allocator in the current
     thread.
     """
-    mkl.mkl_thread_free_buffers()
+    with nogil:
+        mkl.mkl_thread_free_buffers()
     return