根据L2一阶段中调用的L0接口,计算需要的workspace大小。
uint64_t GetWorkspaceSize()
无
返回L2接口在运行中需要的workspace大小。
无
1 2 3 4 5 6 | // aclnn固定写法 void aclnnAddGetWorkspaceSize(..., uint64_t *workspaceSize, aclOpExecutor **executor) { auto uniqueExecutor = CREATE_EXECUTOR(); ...... *workspaceSize = uniqueExecutor->GetWorkspaceSize(); } |