CANN/asc-devkit矩阵计算搬入说明

CANN/asc-devkit矩阵计算搬入说明
矩阵计算的搬入总体说明【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit矩阵计算的搬入接口主要实现Global Memory到L1 Buffer和L1 Buffer到L0A Buffer/L0B Buffer/L0ScaleA Buffer/L0ScaleB Buffer/BiasTable Buffer/Fixpipe Buffer的数据高效传输。针对Ascend 950PR/Ascend 950DT表1数据通路和存储层级源位置源地址对齐要求目的位置目的地址对齐要求格式转换典型应用场景Global Memory1字节L1 Buffer32字节ND2ND、ND2NZ、ND2ZN、DN2NZ、DN2ZN、NZ2NZ、ZN2ZN从全局内存加载左矩阵、右矩阵等数据到L1 Buffer。Global Memory1字节L1 Buffer32字节ScaleND2ZZ、ScaleDN2ZZ、ZZ2ZZ、ScaleND2NN、ScaleDN2NN、NN2NN从全局内存加载左矩阵缩放系数和右矩阵缩放系数到L1 Buffer。L1 Buffer32字节L0A Buffer512字节NZ2NZ、ZN2NZ从L1 Buffer加载左矩阵数据到L0A Buffer。L1 Buffer32字节L0B Buffer512字节ZN2ZN、NZ2ZN从L1 Buffer加载右矩阵数据到L0B Buffer。L1 Buffer32字节L0ScaleA Buffer32字节ZZ2ZZ从L1 Buffer加载左矩阵缩放系数到L0ScaleA Buffer。L1 Buffer32字节L0ScaleB Buffer32字节NN2NN从L1 Buffer加载右矩阵缩放系数到L0ScaleB Buffer。L1 Buffer32字节BiasTable Buffer64字节ND2ND从L1 Buffer加载Bias数据到BiasTable Buffer。L1 Buffer32字节Fixpipe Buffer128字节ND2ND从L1 Buffer加载量化数据到Fixpipe Buffer。【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考