时间戳

首页 > 解决方案 > 从 CMake 获取 BLAS 实现信息

cmake - 从 CMake 获取 BLAS 实现信息

问题描述

在我正在使用的项目中

find_package(BLAS REQUIRED)

检测 BLAS。

有没有办法判断在此之后找到了哪个 BLAS实现？

根据文档 BLA_VENDOR可用于要求某个实现，但它不报告找到了哪一个。

不幸的是，我需要知道找到了哪个 BLAS，因为不同的实现在它们的接口上有细微的差异，例如 MKL 使用zdotu6 个参数而不是 5 个（第一个是指向结果值的指针）。

标签： cmakeblasintel-mklopenblas

解决方案

根据@Tsyvarev 的建议，我最终这样做了

set(BLA_VENDOR Intel10_64lp)
find_package(BLAS)
if(BLAS_FOUND)
    message("MKL environment detected")
    add_definitions(-DRETURN_BY_STACK)
else()
    unset(BLA_VENDOR)
    find_package(BLAS REQUIRED)
endif()

似乎-DRETURN_BY_STACK(or -DFORTRAN_COMPLEX_FUNCTIONS_RETURN_VOID) 是某些BLAS 标头 (例如cblas.h) 识别具有 MKL 兼容声明的一种方式。

从技术上讲，可能需要尝试使用intelmkl BLAS 的每个变体，并以某种方式选择一个，我将其放在这里是为了完整性。这也假设最后一个find_package

Intel10_32 (intel mkl v10 32 bit)
Intel10_64lp (intel mkl v10+ 64 bit, threaded code, lp64 model)
Intel10_64lp_seq (intel mkl v10+ 64 bit, sequential code, lp64 model)
Intel10_64ilp (intel mkl v10+ 64 bit, threaded code, ilp64 model)
Intel10_64ilp_seq (intel mkl v10+ 64 bit, sequential code, ilp64 model)
Intel

推荐阅读