0

0

c++23的std::mdspan如何与BLAS/LAPACK库结合使用? (高性能计算)

冰火之心

冰火之心

发布时间:2026-01-10 13:52:02

|

541人浏览过

|

来源于php中文网

原创

std::mdspan不直接调用BLAS/LAPACK,仅描述布局;需手动提取data_handle()和匹配layout的stride传入函数,列优先时lda=A.stride(1),行优先需转置;LAPACK调用要求可写连续内存、独立pivot数组及检查info。

c++23的std::mdspan如何与blas/lapack库结合使用? (高性能计算)

std::mdspan 本身不直接调用 BLAS/LAPACK

这是最关键的前置判断:std::mdspan 是 C++23 的视图类,只描述多维数据的布局和访问方式,不提供任何计算逻辑。它不能自动绑定到 dgemmdgetrf 等 BLAS/LAPACK 函数。你需要手动提取指针、步长、尺寸,并按目标库要求的内存布局(通常是列优先)做适配。

从 mdspan 提取原始指针和 stride 需谨慎处理 layout

std::mdspandata_handle() 可安全用于获取底层数据起始地址,但 stride 计算必须匹配其 layout_mapping。例如:

  • 若使用 std::layout_left(列优先),mdspan.extent(0) 是行数,mdspan.stride(1) 是列步长,常对应 BLAS 中的 lda
  • 若使用 std::layout_right(行优先),mdspan.stride(0) 才是行步长,此时需转置或重排数据才能高效对接传统 BLAS
  • std::layout_stride 最灵活,但需显式检查 mdspan.stride(i) 是否符合 BLAS 对 leading dimension 的要求

错误假设 layout 类型会导致传入错误的 lda,引发越界或结果错乱 —— 这是最常见的 crash 原因。

典型 BLAS 调用示例:dgemm 封装适配

以下封装假设输入 mdspan 使用 std::layout_left,与 OpenBLAS/Intel MKL 的列优先约定一致:

Sider
Sider

多功能AI浏览器助手,帮助用户进行聊天、写作、阅读、翻译等

下载

立即学习C++免费学习笔记(深入)”;

extern "C" void dgemm_(const char*, const char*, 
                        const int*, const int*, const int*,
                        const double*, const double*, const int*,
                        const double*, const int*, 
                        const double*, double*, const int*);

template
void call_dgemm(std::mdspan, std::layout_left> A,
                std::mdspan, std::layout_left> B,
                std::mdspan, std::layout_left> C) {
    const char transa = 'N', transb = 'N';
    const int m = C.extent(0), n = C.extent(1), k = A.extent(1);
    const T alpha = 1.0, beta = 0.0;
    // A: lda = A.stride(1) == 4 (col-major: leading dim = rows)
    // B: ldb = B.stride(1) == 5
    // C: ldc = C.stride(1) == 4
    dgemm_(&transa, &transb, &m, &n, &k,
           &alpha,
           A.data_handle(), &A.stride(1),
           B.data_handle(), &B.stride(1),
           &beta,
           C.data_handle(), &C.stride(1));
}

与 LAPACK(如 dgetrf)结合时注意 pivots 和 info 输出

LAPACK 函数如 dgetrf_ 会原地修改矩阵并写入 pivot 数组和 info 返回值。适配要点:

  • std::mdspan 必须是可写的(非 const),且底层内存需连续、对齐(建议用 std::vectoraligned_alloc 分配)
  • pivot 数组需独立分配,类型为 int*,长度至少为 min(m,n)
  • info 是标量输出,必须传地址:&info
  • mdspan 使用非标准 layout(如自定义 stride),务必确认 lda 传的是实际行数(列主序下即第一维大小)

忽略 info 检查或误传 lda 会导致数值失败被静默吞掉,后续计算全错却无提示。

相关专题

更多
c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

520

2023.09.20

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

315

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

533

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

51

2025.08.29

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

193

2025.08.29

c++主流开发框架汇总
c++主流开发框架汇总

本专题整合了c++开发框架推荐,阅读专题下面的文章了解更多详细内容。

50

2026.01.09

c++框架学习教程汇总
c++框架学习教程汇总

本专题整合了c++框架学习教程汇总,阅读专题下面的文章了解更多详细内容。

28

2026.01.09

学python好用的网站推荐
学python好用的网站推荐

本专题整合了python学习教程汇总,阅读专题下面的文章了解更多详细内容。

90

2026.01.09

学python网站汇总
学python网站汇总

本专题整合了学python网站汇总,阅读专题下面的文章了解更多详细内容。

10

2026.01.09

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Java 教程
Java 教程

共578课时 | 44.5万人学习

国外Web开发全栈课程全集
国外Web开发全栈课程全集

共12课时 | 1.0万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号