Skip to content

Commit 0a1b1c2

Browse files
Added doc strings
Added asserts Added tests
1 parent d70ac4d commit 0a1b1c2

File tree

4 files changed

+732
-290
lines changed

4 files changed

+732
-290
lines changed

mkl-service/_mkl_service.pxd

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,12 @@ cdef extern from "mkl.h":
7373
# In
7474
int MKL_CBWR_BRANCH
7575
int MKL_CBWR_ALL
76+
# Out
77+
int MKL_CBWR_BRANCH_OFF
7678

7779
# int mkl_enable_instructions(int isa)
7880
# In
81+
int MKL_ENABLE_AVX512_MIC_E1
7982
int MKL_ENABLE_AVX512
8083
int MKL_ENABLE_AVX512_MIC
8184
int MKL_ENABLE_AVX2
@@ -85,7 +88,7 @@ cdef extern from "mkl.h":
8588
int MKL_BLACS_CUSTOM
8689
int MKL_BLACS_MSMPI
8790
int MKL_BLACS_INTELMPI
88-
#int MKL_BLACS_MPICH
91+
int MKL_BLACS_MPICH2
8992

9093
# unsigned int vmlSetMode(unsigned int mode)
9194
# In
@@ -112,6 +115,12 @@ cdef extern from "mkl.h":
112115
int VML_STATUS_OVERFLOW
113116
int VML_STATUS_UNDERFLOW
114117

118+
# unsigned int vmlGetMode()
119+
# Mask
120+
int VML_ACCURACY_MASK
121+
int VML_FTZDAZ_MASK
122+
int VML_ERRMODE_MASK
123+
115124
ctypedef struct MKLVersion:
116125
int MajorVersion
117126
int MinorVersion

0 commit comments

Comments
 (0)