cublas_学习笔记2

由于cublas为了更大的适应Fortan语言，二维数据的存储采用以列优先的方式，这与C/C++中，行优先的存储方式不同。由于本人的研究是数据的来源是C代码得到的，为了加速矩阵的运算效率，利用cublas来完成。本文档提出了一种有效的解决方案。

为了更好的说明，以函数cublasSgemm的实现C= A*B为例。接口cublasSgemm 实现的功能为C = alpha*A*B + beta*C，为了完成C= A*B 的功能，令alpha= 1.0f，beta = 0.0f

利用cublasSgemm的参数 transa（transb）和 lda（ldb）的设置来共同解决存储方式改变的问题。

第一种：输入矩阵A,B均在CPU上

cublasSgemm('t','t',结果矩阵C的行,结果矩阵C的列，左矩阵A的列, alpha，左矩阵,左矩阵的列,右矩阵,右矩阵的列, beta,结果矩阵,结果矩阵的行);

第二种：输入矩阵A在显存上以column-major的方式存储，B在CPU上

cublasSgemm('n','t',结果矩阵C的行,结果矩阵C的列，左矩阵A的列, alpha，左矩阵A,左矩阵A的行,右矩阵B,右矩阵B的列, beta,结果矩阵C,结果矩阵C的行);

第三种：输入矩阵B在显存上以column-major的方式存储，A在CPU上

cublasSgemm('t','n',结果矩阵C的行,结果矩阵C的列，左矩阵A的列, alpha，左矩阵A,左矩阵A的列,右矩阵B,右矩阵B的行, beta,结果矩阵C,结果矩阵C的行);

第四种：输入矩阵A，B均在显存上，以column-major的方式存储

cublasSgemm('n','n',结果矩阵C的行,结果矩阵C的列，左矩阵A的列, alpha，左矩阵A,左矩阵A的行,右矩阵B,右矩阵B的行, beta,结果矩阵C,结果矩阵C的行);

总结：如果前边的参数是't'，那么leading dimesion 就是矩阵的列数，因为此时的矩阵是按照C语言以行优先的方式来存储的；反之如果前边的参数是'n'，那么leading dimesion 就是矩阵的行数，此时的矩阵保持CUBLAS的列优先存储方式；

期待cublas的高手们，分享解决该问题的方法，先谢了~

cublas_学习笔记2相关推荐

PyTorch 学习笔记（六）：PyTorch hook 和关于 PyTorch backward 过程的理解 call
您的位置首页 PyTorch 学习笔记系列 PyTorch 学习笔记(六):PyTorch hook 和关于 PyTorch backward 过程的理解发布: 2017年8月4日 7,195阅读 ...
容器云原生DevOps学习笔记——第三期：从零搭建CI/CD系统标准化交付流程
暑期实习期间,所在的技术中台-效能研发团队规划设计并结合公司开源协同实现符合DevOps理念的研发工具平台,实现研发过程自动化.标准化: 实习期间对DevOps的理解一直懵懵懂懂,最近观看了阿里专家带 ...
容器云原生DevOps学习笔记——第二期：如何快速高质量的应用容器化迁移
暑期实习期间,所在的技术中台-效能研发团队规划设计并结合公司开源协同实现符合DevOps理念的研发工具平台,实现研发过程自动化.标准化: 实习期间对DevOps的理解一直懵懵懂懂,最近观看了阿里专家带 ...
2020年Yann Lecun深度学习笔记（下）
2020年Yann Lecun深度学习笔记(下)
2020年Yann Lecun深度学习笔记（上）
2020年Yann Lecun深度学习笔记(上)
知识图谱学习笔记（1）
知识图谱学习笔记第一部分,包含RDF介绍,以及Jena RDF API使用知识图谱的基石:RDF RDF(Resource Description Framework),即资源描述框架,其本质是一个 ...
计算机基础知识第十讲,计算机文化基础（第十讲）学习笔记
计算机文化基础(第十讲)学习笔记采样和量化PictureElement Pixel(像素)(链接: 采样的实质就是要用多少点(这个点我们叫像素)来描述一张图像,比如,一幅420x570的图像,就表示 ...
Go 学习推荐 —（Go by example 中文版、Go 构建 Web 应用、Go 学习笔记、Golang常见错误、Go 语言四十二章经、Go 语言高级编程）
Go by example 中文版 Go 构建 Web 应用 Go 学习笔记:无痕 Go 标准库中文文档 Golang开发新手常犯的50个错误 50 Shades of Go: Traps, Gotc ...
MongoDB学习笔记(入门)
MongoDB学习笔记(入门) 一.文档的注意事项: 1. 键值对是有序的,如:{ "name" : "stephen", "genda" ...

cublas_学习笔记2

cublas_学习笔记2相关推荐

最新文章

热门文章