Added mkl_cbwr_get funcion

Author: dmitrii-zagornyi

mkl-service/_mkl_service.pxd

@@ -43,6 +43,11 @@ cdef extern from "mkl.h":
     int MKL_CBWR_ERR_UNSUPPORTED_BRANCH
     int MKL_CBWR_ERR_MODE_CHANGE_FAILURE
 
+    # int mkl_cbwr_get(int option)
+    # In
+    int MKL_CBWR_BRANCH
+    int MKL_CBWR_ALL
+
     # int mkl_enable_instructions(int isa)
     # In
     int MKL_ENABLE_AVX512
@@ -117,6 +122,7 @@ cdef extern from "mkl.h":
 
     # Conditional Numerical Reproducibility
     int mkl_cbwr_set(int settings)
+    int mkl_cbwr_get(int option)
     int mkl_cbwr_get_auto_branch()
 
     # Miscellaneous

mkl-service/_mkl_service.pyx

@@ -31,6 +31,11 @@ MKL_CBWR_ERR_INVALID_INPUT = mkl.MKL_CBWR_ERR_INVALID_INPUT
 MKL_CBWR_ERR_UNSUPPORTED_BRANCH = mkl.MKL_CBWR_ERR_UNSUPPORTED_BRANCH
 MKL_CBWR_ERR_MODE_CHANGE_FAILURE = mkl.MKL_CBWR_ERR_MODE_CHANGE_FAILURE
 
+# int mkl_cbwr_get(int option)
+# In
+MKL_CBWR_BRANCH = mkl.MKL_CBWR_BRANCH
+MKL_CBWR_ALL = mkl.MKL_CBWR_ALL
+
 # int mkl_enable_instructions(int isa)
 # In
 MKL_ENABLE_AVX512 = mkl.MKL_ENABLE_AVX512
@@ -197,11 +202,14 @@ def mkl_set_memory_limit(mem_type, limit):
 '''
     #Conditional Numerical Reproducibility
     int mkl_cbwr_set(int settings)
+    int mkl_cbwr_get()
     int mkl_cbwr_get_auto_branch()
 '''
 def mkl_cbwr_set(settings):
     return mkl.mkl_cbwr_set(settings)
 
+def mkl_cbwr_get(option):
+    return mkl.mkl_cbwr_get(option)
 
 def mkl_cbwr_get_auto_branch():
     return mkl.mkl_cbwr_get_auto_branch()
@@ -222,9 +230,9 @@ def mkl_set_env_mode(mode):
     return mkl.mkl_set_env_mode(mode)
 
 
-def mkl_verbose(isEnabled):
-    assert(type(isEnabled) is bool)
-    return mkl.mkl_verbose(isEnabled)
+def mkl_verbose(enable):
+    assert(type(enable) is bool)
+    return mkl.mkl_verbose(enable)
 
 
 def mkl_set_mpi(vendor, custom_library_name):

mkl-service/tests/test_mkl_service.py

@@ -90,6 +90,9 @@ class test_conditional_numerical_reproducibility_control:
     def test_mkl_cbwr_set(self):
         mkl_service.mkl_cbwr_set(mkl_service.MKL_CBWR_AUTO)
 
+    def test_mkl_cbwr_get(self):
+        mkl_service.mkl_cbwr_get(mkl_service.MKL_CBWR_ALL)
+
     def test_mkl_cbwr_get_auto_branch(self):
         mkl_service.mkl_cbwr_get_auto_branch()
 
@@ -103,7 +106,7 @@ def test_mkl_set_env_mode(self):
         mkl_service.mkl_set_env_mode(0)
 
     def test_mkl_verbose(self):
-        mkl_service.mkl_verbose(0)
+        mkl_service.mkl_verbose(False)
 
     def test_mkl_set_mpi(self):
         mkl_service.mkl_set_mpi(1, 'test')