Skip to content

Commit e76c8a1

Browse files
Implemented support for CBWR_STRICT
mkl.cbwr_set('avx2,strict') mkl.cbwr_set('avx512,strict') Also added support for missing constants mkl.cbwr_set('avx512_e1'), mkl.cbwr_set('avx512_mic_e1') on appropriate architectures.
1 parent f14d7dd commit e76c8a1

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

mkl/_mkl_service.pxd

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,9 @@ cdef extern from "mkl.h":
5656
int MKL_CBWR_AVX2
5757
int MKL_CBWR_AVX512_MIC
5858
int MKL_CBWR_AVX512
59+
int MKL_CBWR_STRICT
60+
int MKL_CBWR_AVX512_E1
61+
int MKL_CBWR_AVX512_MIC_E1
5962
int MKL_CBWR_BRANCH
6063
int MKL_CBWR_ALL
6164
int MKL_CBWR_SUCCESS

mkl/_mkl_service.pyx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -653,6 +653,12 @@ cdef object __cbwr_set(branch=None):
653653
'avx2': mkl.MKL_CBWR_AVX2,
654654
'avx512_mic': mkl.MKL_CBWR_AVX512_MIC,
655655
'avx512': mkl.MKL_CBWR_AVX512,
656+
'avx512_e1': mkl.MKL_CBWR_AVX512_E1,
657+
'avx512_mic_e1': mkl.MKL_CBWR_AVX512_MIC_E1,
658+
'avx2,strict': mkl.MKL_CBWR_AVX2 | mkl.MKL_CBWR_STRICT,
659+
'avx512_mic,strict': mkl.MKL_CBWR_AVX512_MIC | mkl.MKL_CBWR_STRICT,
660+
'avx512,strict': mkl.MKL_CBWR_AVX512 | mkl.MKL_CBWR_STRICT,
661+
'avx512_e1,strict': mkl.MKL_CBWR_AVX512_E1 | mkl.MKL_CBWR_STRICT,
656662
},
657663
'output': {
658664
mkl.MKL_CBWR_SUCCESS: 'success',

0 commit comments

Comments
 (0)