asc.language.adv.Matmul.set_org_shape【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyascMatmul.set_org_shape(org_m: int, org_n: int, org_ka: int, org_kb: int | None None, org_kc: int | None None) → None设置Matmul计算原始完整的形状M、N、K单位为元素个数。用于运行时修改shape比如复用同一个Matmul对象从不同的矩阵块取数据计算。对应的Ascend C函数原型__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)参数说明org_m设置原始完整的形状M大小单位为元素。org_n设置原始完整的形状N大小单位为元素。org_ka设置矩阵A原始完整的形状Ka大小单位为元素。org_kb设置矩阵B原始完整的形状Kb大小单位为元素。org_kc设置输出C矩阵的N单位为元素。需要输入B矩阵的N和输出C矩阵的N不一样时可设置默认为0即使用B矩阵的N不进行修改。备注Ascend C第一个函数原型对应的python参数org_morg_norg_kaAscend C第二个函数原型对应的python参数org_morg_norg_kaorg_kborg_kc。约束说明本接口需要在set_tensor_a接口、set_tensor_b接口及set_single_shape接口前调用。调用示例asc.adv.register_matmul(pipe, workspace, mm, tiling) mm.set_tensor_a(gm_a) mm.set_tensor_b(gm_b) mm.set_bias(gm_bias) mm.iterate_all(gm_c) # 复用mm对象 mm.set_org_shape(org_m, org_n, org_k) mm.set_tensor_a(gm_a1) mm.set_tensor_b(gm_b1) mm.set_bias(gm_bias1) mm.iterate_all(gm_c1)【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyasc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考