Added off parameter into mkl_cbwr_set function

Author: dmitrii-zagornyi

mkl-service/_mkl_service.pyx

@@ -617,6 +617,7 @@ cdef inline __cbwr_set(branch=None):
     """
     __variables = {
         'input': {
+            'off': mkl.MKL_CBWR_BRANCH_OFF,
             'auto': mkl.MKL_CBWR_AUTO,
             'compatible': mkl.MKL_CBWR_COMPATIBLE,
             'sse2': mkl.MKL_CBWR_SSE2,
@@ -655,6 +656,7 @@ cdef inline __cbwr_get(cnr_const=None):
             'all': mkl.MKL_CBWR_ALL,
         },
         'output': {
+            mkl.MKL_CBWR_BRANCH_OFF: 'off',
             mkl.MKL_CBWR_AUTO: 'auto',
             mkl.MKL_CBWR_COMPATIBLE: 'compatible',
             mkl.MKL_CBWR_SSE2: 'sse2',
@@ -667,7 +669,6 @@ cdef inline __cbwr_get(cnr_const=None):
             mkl.MKL_CBWR_AVX512_MIC: 'avx512_mic',
             mkl.MKL_CBWR_AVX512: 'avx512',
             mkl.MKL_CBWR_SUCCESS: 'success',
-            mkl.MKL_CBWR_BRANCH_OFF: 'branch_off',
             mkl.MKL_CBWR_ERR_INVALID_INPUT: 'err_invalid_input',
         },
     }

mkl-service/tests/test_mkl_service.py

@@ -159,6 +159,9 @@ def test_set_memory_limit(self):
 
 class test_conditional_numerical_reproducibility_control:
     # https://software.intel.com/en-us/mkl-developer-reference-c-conditional-numerical-reproducibility-control
+    def test_cbwr_set_off(self):
+        mkl.cbwr_set(branch='off')
+
     def test_cbwr_set_auto(self):
         mkl.cbwr_set(branch='auto')