问使用OpenMP拆分LAPACK调用

EN

你能做的最接近的事情就是让一个由两个线程组成的团队执行一个OpenMP并行区域，然后从每个线程调用MKL。您必须在MKL中启用嵌套并行(通过禁用动态线程)，将MKL线程的数量固定为6，并且必须使用英特尔的编译器套件来编译代码。MKL本身是使用OpenMP线程的，但它是英特尔的OpenMP运行时。如果您碰巧使用另一种编译器，例如GCC，它的OpenMP运行时可能会被证明与英特尔的不兼容。

由于您没有指定语言，我提供了两个示例-一个用Fortran，一个用C/C++：

Fortran：

call mkl_set_num_threads(6)
call mkl_set_dynamic(0)

!$omp parallel sections num_threads(2)
!$omp section
   call dgemm(...)
!$omp end section
!$omp section
   call dgemm(...)
!$omp end section
!$omp end parallel sections

C/C++：

mkl_set_num_threads(6);
mkl_set_dynamic(0);

#pragma omp parallel sections num_threads(2)
{
    #pragma omp section
    {
        cblas_dgemm(...)
    }
    #pragma omp section
    {
        cblas_dgemm(...)
    }
}

一般来说，你不能为MKL创建线程的子集(至少在我目前的理解中是这样)。每个DGEMM调用将使用全局指定数量的MKL线程。请注意，MKL操作可能会针对CPU的缓存大小进行调整，并且并行执行两个矩阵乘法可能没有好处。如果您的NUMA系统有两个六边形CPU，每个CPU都有自己的内存控制器(我怀疑这就是您的情况)，那么您可能会这样做，但是您必须注意数据放置的位置，并启用线程到内核的绑定(锁定)。