MmadTraits

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√

功能说明

MmadTraits用于定义矩阵乘加操作的特性，包括矩阵乘加操作类型和相关参数。

结构体定义

template <typename MadOperation, typename... MadOpArgs>
struct MmadTraits {
    using OperationType = MadOperation;
    std::tuple<MadOpArgs...> args;
};

字段说明

字段名	类型	描述
OperationType	MadOperation	矩阵乘加操作类型。
args	std::tuple<MadOpArgs...>	矩阵乘加操作参数。

约束说明

• MadOperation必须是有效的矩阵乘加操作类型。
• MadOpArgs的数量和类型必须与MadOperation的要求匹配。

调用示例

using namespace AscendC::Te;
// 创建MmadTraits
using mmadTraits = MmadTraits<MmadOperation, MmadTraitDefault>;
// 使用mmadTraits声明MmadAtom对象，调用Mmad接口实现MmadOperation，其中l0CTensor是位于L0C上的计算结果矩阵，l0ATensor是位于L0A上的左矩阵，l0BTensor是位于L0B上的右矩阵，para是Mmad计算运行时参数。
Mmad(MmadAtom<mmadTraits>{}, l0CTensor, l0ATensor, l0BTensor, para);