如何解决如何更好地控制 CPU 上每个 BLAS 内核调用所使用的线程数?
我正在编写一个 OpenMP 代码,它在不同的线程中调用不同的 BLAS 内核,主要是具有不同大小的 DGEMM。为了最大限度地提高性能,我想控制我为每个 BLAS 调用的线程数。 看起来这是一个很明显的基本需求,虽然很难做到。
OpenBLAS有一个函数openblas_set_num_threads(int n)
,在OpenBLAS代码的README文件中是这样描述的
这些只在库初始化时使用一次,不可用于 微调各个 BLAS 调用中的线程数。
所以我想我不能在 OpenBLAS 中使用这个功能。
MKL 有一个函数 mkl_set_num_threads_local(int nt)
,这似乎是我在使用 MKL 时的问题的答案。
有没有一种方法可以为每个 BLAS 调用调整线程数,而不管我使用的是什么库? (理想的选择) 如果没有,是不是只有 MKL 才能让我调整线程数?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。