From d04a277016ec4af4e507131a3751daca028edc1d Mon Sep 17 00:00:00 2001 From: asistradition Date: Mon, 23 Aug 2021 17:28:05 -0400 Subject: [PATCH] Add Num Threads for Debug Message --- sparse_dot_mkl/_mkl_interface.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/sparse_dot_mkl/_mkl_interface.py b/sparse_dot_mkl/_mkl_interface.py index aed4868..a7c5c7c 100644 --- a/sparse_dot_mkl/_mkl_interface.py +++ b/sparse_dot_mkl/_mkl_interface.py @@ -44,7 +44,7 @@ # Since it's not on PyPi I don't want to make this an actual package dependency # So without it just create mock functions and don't do version checking try: - from mkl import get_version, get_version_string + from mkl import get_version, get_version_string, get_max_threads except ImportError: def get_version(): return None @@ -53,6 +53,10 @@ def get_version(): def get_version_string(): return None + + def get_max_threads(): + return None + if get_version() is not None and get_version()["MajorVersion"] < 2020: _verr_msg = "Loaded version of MKL is out of date: {v}".format(v=get_version_string()) warnings.warn(_verr_msg) @@ -518,6 +522,7 @@ def print_mkl_debug(): print("mkl-service must be installed to get full debug messaging") else: print(get_version_string()) + print("MKL Number of Threads: {n}".format(n=get_max_threads())) print("MKL linked: {fn}".format(fn=_libmkl._name)) print("MKL interface {np} | {c}".format(np=MKL.MKL_INT_NUMPY, c=MKL.MKL_INT))