CANN/catlass 卷积瓦片拷贝

ConvTileCopy

【免费下载链接】catlass本项目是CANN的算子模板库，提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlass

代码位置

[TOC]

功能说明

ConvTileCopy是卷积（Conv）场景的搬运模板集合，结构上与 TileCopy 完全相同，区别仅在于模板参数命名和用途——专门服务于 Convolution 的 Im2Col + GEMM 流程。

BiasType 为必选参数（非默认void），因为 Conv 通常包含 Bias。

引用的 Tile 组件

模板原型

template < class ArchTag, // 架构标签 class AType, // A 矩阵 GmType class BType, // B 矩阵 GmType class CType, // C 矩阵 GmType class BiasType // Bias GmType（必选） > struct ConvTileCopy;

调用示例

#include "catlass/gemm/tile/tile_copy.hpp" using namespace Catlass::Gemm; using AType = Gemm::GemmType<half, layout::RowMajor>; using BType = Gemm::GemmType<half, layout::ColumnMajor>; using CType = Gemm::GemmType<half, layout::NDC1HWC0>; using BiasType = Gemm::GemmType<half, layout::VectorLayout>; using TileCopy_ = Tile::ConvTileCopy<Arch::AtlasA2, AType, BType, CType, BiasType>; typename TileCopy_::CopyGmToL1A copyGmToL1A; typename TileCopy_::CopyGmToL1B copyGmToL1B; typename TileCopy_::CopyGmToL1Bias copyGmToL1Bias; typename TileCopy_::CopyL1ToBT copyL1ToBT; typename TileCopy_::CopyL0CToGm copyL0CToGm;

【免费下载链接】catlass本项目是CANN的算子模板库，提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlass