mkl 服务

这个包公开了一些在 mkl_service.h 中声明的函数。该包的主要目的是允许用户在运行时更改 MKL 使用的 CPU 数量。

例子:

在这个小例子中,我们首先查询正在使用多少 CPU,然后更改这个数字。

>>> import mkl
>>> mkl.get_max_threads()
2
>>> mkl.set_num_threads(1)
>>> mkl.get_max_threads()
1

参考

mkl服务功能:

mkl.get_cpu_clocks() -> 整数
以整数形式返回 CPU 时钟。
mkl.get_cpu_frequency() -> 浮动
以 GHz 为单位返回 CPU 频率作为浮点数。
mkl.get_max_threads() -> 整数
返回英特尔 MKL 针对并行性的线程数。
mkl.get_version_string() -> 字符串
以字符串形式返回 MKL 库版本信息。
mkl.mem_stat(n) -> 整数
返回由 MKL 内存分配器分配的内存量。
mkl.set_num_threads(n)
设置 MKL 应使用的线程数。这只是一个提示,并不能保证实际使用这个线程数。此函数优先于环境变量 MKL_NUM_THREADS。