--- title: SetOrgShape description: "| 产品 | 是否支持 |" url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendcopapi/atlasascendc_api_07_0651.html sourcePath: /source/zh/canncommercial/900/API/ascendcopapi/atlasascendc_api_07_0651.html indexId: eb6e6402c4b64efb34d4c1ab77448434c6181c77dd75a2711f67018eef20cacf76 --- # SetOrgShape #### 产品支持情况 | 产品 | 是否支持 | | --- | --- | | Atlas 350 加速卡 | √ | | Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ | | Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ | | Atlas 200I/500 A2 推理产品 | √ | | Atlas 推理系列产品 AI Core | √ | | Atlas 推理系列产品 Vector Core | x | | Atlas 训练系列产品 | x | #### 功能说明 设置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) ``` #### 参数说明 **表1 参数说明** | 参数名 | 输入/输出 | 描述 | | --- | --- | --- | | orgM | 输入 | 设置原始完整的形状M大小，单位为元素。 对于Atlas 350 加速卡，在使用MDL模板创建的Matmul对象调用本接口时，该参数用于设置GM或L1上完整的形状M大小，单位为元素。 | | orgN | 输入 | 设置原始完整的形状N大小，单位为元素。 对于Atlas 350 加速卡，在使用MDL模板创建的Matmul对象调用本接口时，该参数用于设置GM或L1上完整的形状N大小，单位为元素。 | | orgK | 输入 | 设置原始完整的形状K大小，单位为元素。原始完整形状Ka=Kb时可设置。 对于Atlas 350 加速卡，在使用MDL模板创建的Matmul对象调用本接口时，该参数用于设置GM或L1上完整的形状K大小，单位为元素。原始完整形状Ka=Kb时可设置。 | | orgKa | 输入 | 设置矩阵A原始完整的形状Ka大小，单位为元素。 对于Atlas 350 加速卡，在使用MDL模板创建的Matmul对象调用本接口时，该参数用于设置GM或L1上完整的形状Ka大小，单位为元素。 | | orgKb | 输入 | 设置矩阵B原始完整的形状Kb大小，单位为元素。 对于Atlas 350 加速卡，在使用MDL模板创建的Matmul对象调用本接口时，该参数用于设置GM或L1上完整的形状Kb大小，单位为元素。 | | orgKc | 输入 | 设置输出C矩阵的N，单位为元素。需要输入B矩阵的N和输出C矩阵的N不一样时可设置，默认为0（即使用B矩阵的N，不进行修改）。 | - 对于Atlas 350 加速卡上使用 MDL模板 创建的Matmul对象，L1上数据的形状与Tiling侧接口 SetOrgShape 中的orgMIn/orgNIn/orgKIn/orgKaIn/orgKbIn一致时，不必须调用此接口。 - 对于Atlas 350 加速卡上使用 MDL模板 创建的Matmul对象，L1上数据的形状与Tiling侧接口 SetOrgShape 中的orgMIn/orgNIn/orgKIn/orgKaIn/orgKbIn不一致时，必须调用本接口指定GM/L1上的orgM/orgN/orgK/orgKa/orgKb。 例如，使用MDL模板时，输入矩阵A在L1、输入矩阵B在GM的场景，L1上A的形状大小与Tiling侧原始的orgMIn/orgKIn/orgKaIn不一致时，调用SetOrgShape(orgM, orgN, orgK)/SetOrgShape(orgM, orgN, orgKa, orgKb)接口指定L1上A矩阵相关参数orgM/orgK/orgKa。 #### 返回值说明 无 #### 约束说明 本接口需要在SetTensorA接口、SetTensorB接口、SetBias接口及SetSingleShape接口前调用。 #### 调用示例 - 设置矩阵原始完整的形状 1 2 3 4 5 6 7 8 9 10 11 REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling); mm.SetTensorA(gm_a); mm.SetTensorB(gm_b); mm.SetBias(gm_bias); mm.IterateAll(gm_c); // 复用mm对象 mm.SetOrgShape(orgM, orgN, orgK); mm.SetTensorA(gm_a1); mm.SetTensorB(gm_b1); mm.SetBias(gm_bias1); mm.IterateAll(gm_c1); - 对于Atlas 350 加速卡上使用 MDL模板 的Matmul对象，设置GM或L1上完整的形状 完整的样例请参考matmul_tscm_mdl_setorgshape样例(https://gitcode.com/cann/asc-devkit/tree/9.0.0/examples/01_simd_cpp_api/03_libraries/00_matrix/matmul_tscm_mdl_setorgshape)。 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling); for (int m = 0; m < mIter_; m++) { for (int n = 0; n < nIter_; n++) { for (int k = 0; k < kIter_; k++) { // 复用mm，指定A在L1和B在GM上的shape mm.SetOrgShape(alignedSingleM, tiling.N, alignedSingleK, tiling.Kb, tiling.N); mm.SetSingleShape(curBaseM, curBaseN, curBaseK); mm.SetTensorA(tscm_a[offset_a]); // Set aMatrix tscm input mm.SetTensorB(gm_b[offset_b]); mm.SetBias(gm_bias[offset_bias]); mm.Iterate(k != 0); } matmulObj.GetTensorC(gm_c[offset_c]); } }