你如何获得*Julia中BLAS操作的线程数？

Question

Julia的blas库有一个set_num_threads()为BLAS操作设置线程的函数.

如何获取默认使用的线程数,或者当前？

Answer 1

如果您将Julia链接到OpenBLAS,如果您使用来自julialang.org的Windows,Mac或通用Linux二进制文件,或者如果您从源代码构建并且未设置USE_SYSTEM_BLAS=1,那么您将获得所使用的数字线程通过调用OpenBLAS ccall((:openblas_get_num_threads64_, Base.libblas_name), Cint, ()).最终,我们可能会将其作为Julia函数提供.

Answer 2

从 Julia 1.6 开始，您可以使用LinearAlgebra.BLAS.get_num_threads(). 所以下面的代码就可以解决这个问题：

using LinearAlgebra
BLAS.get_num_threads()