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。