CANN/catlass GEMM块API文档

Gemm/Block Class Template Overview【免费下载链接】catlass本项目是CANN的算子模板库提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlassAPI ListblockMmad ListComponentDescriptionblock_mmadBasic template, including BlockMmad.block_mmad_pingpongPartial specialization of BlockMmad implementing ping-pong matrix multiplication.Swizzle ListComponentDescriptionblock_swizzleBasic swizzle methodsGemmIdentityBlockSwizzleBasic swizzle policy for the GEMM operatorAPI BreakdownblockMmadThe blockMmad structure encapsulates the MMAD computation at the Block layer, mapping directly to execution on a single AI Core of the Ascend NPU. Through template parameters, it receives configuration details defining the matrix shapes, tensor layouts (such as row-major or column-major), and data types (DType).The namespace isCatlass::Gemm::Block. Core members:TypeNameFunctionConstructorBlockMmad()Initializes buffers, registers event IDs, and insertssetFlagprimitives for pipeline synchronization.Destructor~BlockMmad()InsertswaitFlagprimitives for pipeline synchronization.Functionvoid operator()Executes the matrix multiplication for a Block task.【免费下载链接】catlass本项目是CANN的算子模板库提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlass创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考