Skip to content

Commit 1454d1b

Browse files
Changed grouping and added comments for constants
1 parent 4757b68 commit 1454d1b

File tree

1 file changed

+9
-28
lines changed

1 file changed

+9
-28
lines changed

mkl-service/_mkl_service.pxd

Lines changed: 9 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -30,28 +30,21 @@ ctypedef int MKL_INT
3030

3131

3232
cdef extern from "mkl.h":
33-
# int mkl_domain_set_num_threads(int nt, int domain)
34-
# int mkl_domain_get_max_threads(int domain)
35-
# In
33+
# MKL Function Domains Constants
3634
int MKL_DOMAIN_BLAS
3735
int MKL_DOMAIN_FFT
3836
int MKL_DOMAIN_VML
3937
int MKL_DOMAIN_PARDISO
4038
int MKL_DOMAIN_ALL
4139

42-
# MKL_INT64 mkl_peak_mem_usage(int mode)
43-
# In
40+
# MKL Peak Memory Usage Constants
4441
int MKL_PEAK_MEM_ENABLE
4542
int MKL_PEAK_MEM_DISABLE
4643
int MKL_PEAK_MEM
4744
int MKL_PEAK_MEM_RESET
48-
49-
# int mkl_set_memory_limit(int mem_type, size_t limit)
50-
# In
5145
int MKL_MEM_MCDRAM
5246

53-
# int mkl_cbwr_set(int settings)
54-
# In
47+
# CNR Control Constants
5548
int MKL_CBWR_AUTO
5649
int MKL_CBWR_COMPATIBLE
5750
int MKL_CBWR_SSE2
@@ -63,35 +56,29 @@ cdef extern from "mkl.h":
6356
int MKL_CBWR_AVX2
6457
int MKL_CBWR_AVX512_MIC
6558
int MKL_CBWR_AVX512
66-
# Out
59+
int MKL_CBWR_BRANCH
60+
int MKL_CBWR_ALL
6761
int MKL_CBWR_SUCCESS
62+
int MKL_CBWR_BRANCH_OFF
6863
int MKL_CBWR_ERR_INVALID_INPUT
6964
int MKL_CBWR_ERR_UNSUPPORTED_BRANCH
7065
int MKL_CBWR_ERR_MODE_CHANGE_FAILURE
7166

72-
# int mkl_cbwr_get(int option)
73-
# In
74-
int MKL_CBWR_BRANCH
75-
int MKL_CBWR_ALL
76-
# Out
77-
int MKL_CBWR_BRANCH_OFF
78-
79-
# int mkl_enable_instructions(int isa)
80-
# In
67+
# ISA Constants
8168
int MKL_ENABLE_AVX512_MIC_E1
8269
int MKL_ENABLE_AVX512
8370
int MKL_ENABLE_AVX512_MIC
8471
int MKL_ENABLE_AVX2
8572
int MKL_ENABLE_AVX
8673
int MKL_ENABLE_SSE4_2
8774

75+
# MPI Implementation Constants
8876
int MKL_BLACS_CUSTOM
8977
int MKL_BLACS_MSMPI
9078
int MKL_BLACS_INTELMPI
9179
int MKL_BLACS_MPICH2
9280

93-
# unsigned int vmlSetMode(unsigned int mode)
94-
# In
81+
# VML Constants
9582
int VML_HA
9683
int VML_LA
9784
int VML_EP
@@ -103,9 +90,6 @@ cdef extern from "mkl.h":
10390
int VML_ERRMODE_EXCEPT
10491
int VML_ERRMODE_CALLBACK
10592
int VML_ERRMODE_DEFAULT
106-
107-
# int vmlSetErrStatus(const MKL_INT status)
108-
# In
10993
int VML_STATUS_OK
11094
int VML_STATUS_ACCURACYWARNING
11195
int VML_STATUS_BADSIZE
@@ -114,9 +98,6 @@ cdef extern from "mkl.h":
11498
int VML_STATUS_SING
11599
int VML_STATUS_OVERFLOW
116100
int VML_STATUS_UNDERFLOW
117-
118-
# unsigned int vmlGetMode()
119-
# Mask
120101
int VML_ACCURACY_MASK
121102
int VML_FTZDAZ_MASK
122103
int VML_ERRMODE_MASK

0 commit comments

Comments
 (0)