oneMKL无法通过OpenMP卸载至Intel GPU的问题排查求助

运行OneAPI官方示例代码时，发现以下代码并未实际在GPU上执行：

#pragma omp target data map(to:a[0:sizea],b[0:sizeb]) map(tofrom:c[0:sizec]) device(dnum)
{
    // run gemm on gpu, use standard oneMKL interface within a variant dispatch construc
    #pragma omp target variant dispatch device(dnum) use_device_ptr(a, b, c)
    {
        cblas_zgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc);
    }
}

通过设置export LIBOMPTARGET_PLUGIN_PROFILE=T，发现程序运行时无内核耗时；设置export MKL_VERBOSE=1后，显示MKL函数在GPU上的运行次数为0。Linux平台搭载Intel(R) Graphics显卡，需排查问题原因并解决。

• oneMKL默认后端未指向GPU：oneMKL默认优先使用CPU计算，若未显式指定GPU后端，即使在OpenMP目标区域内调用，仍会回退到CPU执行。
• OpenMP目标区域配置不完整：variant dispatch指令的使用可能缺少编译支持，或device(dnum)指定的设备编号无效，导致设备指针未正确传递给MKL。
• 编译选项缺失：编译时未启用Intel GPU的OpenMP目标支持，或未链接oneMKL的GPU相关运行时库，导致程序无法识别GPU设备。
• 系统环境配置问题：Intel GPU驱动未正确安装或版本过低，OpenMP目标设备插件未加载，导致GPU设备无法被程序识别。

1. 显式指定oneMKL GPU后端 #

• 环境变量方式：运行程序前设置环境变量，强制MKL使用GPU后端：

  export MKL_BLAS=GPU
  export MKL_LAPACK=GPU
  

  （针对Intel Xe架构显卡，若为旧款集成显卡，可尝试export MKL_ENABLE_INSTRUCTIONS=AVX512_MIC_E1）
• 代码内设置：在调用cblas_zgemm前，添加以下代码（需包含mkl_service.h头文件）：

  #include "mkl_service.h"
  mkl_set_backend(MKL_BACKEND_GPU);
  mkl_set_threading_layer(MKL_THREADING_OMP);
  

2. 修正OpenMP编译与运行配置 #

• 编译选项：使用Intel oneAPI的icx/icpx或dpcpp编译器时，添加以下编译选项以支持Intel GPU的OpenMP目标：

  icx -fiopenmp -fopenmp-targets=spir64 -o gemm_example gemm_example.c -lmkl_rt -lsycl
  

• 验证设备编号：确认dnum是有效的GPU设备编号，可在代码中添加设备检测逻辑：

  int num_devices = omp_get_num_devices();
  printf("Number of devices: %d\n", num_devices);
  // 确保dnum在0到num_devices-1范围内
  

3. 确认编译链接依赖 #

• 确保已加载oneAPI环境：运行oneAPI的环境配置脚本（如/opt/intel/oneapi/setvars.sh），自动配置库路径与编译选项。
• 编译时链接必要的库：手动编译需链接oneMKL和SYCL相关库，示例链接选项：

  -lmkl_intel_ilp64 -lmkl_sequential -lmkl_core -lmkl_sycl -lsycl -lOpenCL -lpthread -lm
  

4. 检查系统GPU环境 #

• 驱动检查：安装Intel官方GPU驱动，确保版本符合oneAPI要求（建议>=21.38），可通过以下命令查看驱动版本：

  glxinfo | grep "OpenGL version"
  

• 设备识别：运行clinfo命令，确认输出中包含Intel(R) Graphics设备，且OpenCL版本>=2.0。
• OpenMP目标插件：检查LIBOMPTARGET_PLUGIN环境变量是否指向正确的插件文件，示例：

  export LIBOMPTARGET_PLUGIN=/opt/intel/oneapi/compiler/latest/linux/lib/libomptarget.rtl.spir64.so
  

内容的提问来源于stack exchange，提问作者nilv