昇腾社区首页
中文
注册
开发者
下载

两段式接口

基于单算子API执行方式调用算子API时,通常分为“两段式”,样式形如:

[object Object]

必须先调用第一段接口aclxxXxxGetWorkspaceSize,用于计算本次API调用过程中需要多少workspace内存,获取到计算所需的workspaceSize后,按照workspaceSize申请NPU内存,然后调用第二段接口aclxxXxx执行计算。

其中“aclxx”表示算子接口前缀,如aclnn;而“Xxx”表示对应的算子类型,如Add算子。

[object Object]