后传-GEMM通用矩阵乘优化

发布时间:2024年01月12日

写在前面

本文不是GEMM优化教程!!!

只适合已经学过GEMM理论,但是在实践过程中对thredidx,blockidx与矩阵中数据的对应关系还有困惑的同学。

可以先看完其他教程再来本文巩固一下。

主体

本文以CUDA矩阵乘法的优化中的gemm_128x128为示例画出线程与数据中的对应关系,先看完链接中的内容再来看这个比较好。

在这里插入图片描述
这里压缩画质了,可以访问link下载

待补充

此示例中m,n,k必须是32的倍数,无法计算通用的矩阵,后续会添加,敬请期待。

写在后面

这是作者第一次写文章,明明脑子里面想的很好,但就是表达不出来,写下来的要么唠唠叨叨一大堆,要么十分简洁,只有自己才能看懂。从小到大最害怕的就是写文章了┭┮﹏┭┮。

作者水平疏浅,还请见谅。如果您在阅读的过程中发现错误或者觉得哪里还需要完善,欢迎联系作者。

文章来源:https://blog.csdn.net/weixin_44270070/article/details/135549258
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。