Skip to content

Commit 09ee83a

Browse files
Added enums with domains for threading control functions
1 parent 934c2bd commit 09ee83a

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

mkl-service/_mkl_service.pxd

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,15 @@ ctypedef int MKL_INT
44

55

66
cdef extern from "mkl.h":
7+
# int mkl_domain_set_num_threads(int nt, int domain)
8+
# int mkl_domain_get_max_threads(int domain)
9+
# In
10+
int MKL_DOMAIN_BLAS
11+
int MKL_DOMAIN_FFT
12+
int MKL_DOMAIN_VML
13+
int MKL_DOMAIN_PARDISO
14+
int MKL_DOMAIN_ALL
15+
716
# MKL_INT64 mkl_peak_mem_usage(int mode)
817
# In
918
int MKL_PEAK_MEM_ENABLE

0 commit comments

Comments
 (0)