CANN单算子API执行接口一般定义为“两段式接口”,以NN算子接口定义为例:
[object Object]
其中aclnnXxxGetWorkspaceSize为第一段接口,主要用于计算本次API调用过程中需要多少的workspace内存,获取到本次API计算所需的workspaceSize后,按照workspaceSize申请AI处理器内存,然后调用第二段接口aclnnXxx执行计算。
[object Object]
CANN单算子API执行接口一般定义为“两段式接口”,以NN算子接口定义为例:
其中aclnnXxxGetWorkspaceSize为第一段接口,主要用于计算本次API调用过程中需要多少的workspace内存,获取到本次API计算所需的workspaceSize后,按照workspaceSize申请AI处理器内存,然后调用第二段接口aclnnXxx执行计算。
[object Object]