CANN/pyasc矩阵乘法工作空间设置

asc.language.adv.Matmul.set_workspace【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyascMatmul.set_workspace(addr: GlobalTensor | GlobalAddress, size: PlainValue | int | None None) → NoneIterate计算的异步场景调用本接口申请一块临时空间来缓存计算结果然后调用GetTensorC时会在该临时空间中获取C的矩阵分片。 IterateNBatch计算时调用本接口申请一块临时空间来缓存计算结果然后根据同步或异步场景进行其它接口的调用。对应的Ascend C函数原型template class T __aicore__ inline void SetWorkspace(GlobalTensorT addr)template class T __aicore__ inline void SetWorkspace(__gm__ const T* addr, int size)参数说明addr用户传入的GM上的workspace空间GlobalTensor类型。addr用户传入的GM上的workspace空间GM地址类型。size传入GM地址时需要配合传入元素个数。约束说明当使能MixDualMaster双主模式场景时即模板参数enableMixDualMaster设置为true不支持使用该接口。调用示例asc.adv.register_matmul(pipe, workspace, mm, tiling) mm.set_workspace(workspace_gm) mm.set_tensor_a(gm_a) mm.set_tensor_b(gm_b) mm.set_bias(gm_bias) mm.iterate(syncTrue) for i in range(single_corem // base_m * single_core_n // base_n): mm.get_tensor_c(tensorgm_c, syncFalse)【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyasc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考