|
|
@ -95,7 +95,8 @@ Status MemcpyAsyncTaskInfo::Distribute() {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Status MemcpyAsyncTaskInfo::CalculateArgs(const domi::TaskDef &task_def, DavinciModel *davinci_model) {
|
|
|
|
Status MemcpyAsyncTaskInfo::CalculateArgs(const domi::TaskDef &task_def, DavinciModel *davinci_model) {
|
|
|
|
OpDescPtr op_desc = davinci_model_->GetOpByIndex(task_def.memcpy_async().op_index());
|
|
|
|
GE_CHECK_NOTNULL(davinci_model);
|
|
|
|
|
|
|
|
OpDescPtr op_desc = davinci_model->GetOpByIndex(task_def.memcpy_async().op_index());
|
|
|
|
// the num of src and dst size is 2
|
|
|
|
// the num of src and dst size is 2
|
|
|
|
uint32_t args_size = sizeof(void *) * 2;
|
|
|
|
uint32_t args_size = sizeof(void *) * 2;
|
|
|
|
args_offset_ = davinci_model->GetTotalArgsSize();
|
|
|
|
args_offset_ = davinci_model->GetTotalArgsSize();
|
|
|
|