SetOrgShape
功能说明
设置Matmul计算原始完整的形状M、N、K,单位为元素个数。用于运行时修改shape,比如复用同一个Matmul对象,从不同的矩阵块取数据计算。
函数原型
__aicore__ inline void SetOrgShape(int orgM, int orgN, int orgK)
__aicore__ inline void SetOrgShape(int orgM, int orgN, int orgKa, int orgKb, int orgKc = 0)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
orgM |
输入 |
设置原始完整的形状M大小,单位为元素。 |
orgN |
输入 |
设置原始完整的形状N大小,单位为元素。 |
orgK |
输入 |
设置原始完整的形状K大小,单位为元素。原始完整形状Ka=Kb时可设置。 |
orgKa |
输入 |
设置矩阵A原始完整的形状Ka大小,单位为元素。 |
orgKb |
输入 |
设置矩阵B原始完整的形状Kb大小,单位为元素。 |
orgKc |
输入 |
设置输出C矩阵的N,单位为元素。需要输入B矩阵的N和输出C矩阵的N不一样时可设置,默认为0(即使用B矩阵的N,不进行修改)。 |
返回值
无
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas推理系列产品AI Core
Atlas 200I/500 A2推理产品
注意事项
需要在SetTensorA和SetTensorB前调用。
父主题: Matmul