|
@@ -112,8 +112,9 @@ Status OpTask::GetProfilingArgs(std::string &model_name, std::string &op_name, u |
|
|
Status OpTask::UpdateRunInfo(const vector<GeTensorDesc> &input_desc, const vector<GeTensorDesc> &output_desc) { |
|
|
Status OpTask::UpdateRunInfo(const vector<GeTensorDesc> &input_desc, const vector<GeTensorDesc> &output_desc) { |
|
|
return UNSUPPORTED; |
|
|
return UNSUPPORTED; |
|
|
} |
|
|
} |
|
|
Status OpTask::UpdateArgTable(const SingleOpModelParam ¶m) { |
|
|
|
|
|
auto addresses = BuildTaskUtils::GetAddresses(op_desc_, param); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Status OpTask::DoUpdateArgTable(const SingleOpModelParam ¶m, bool keep_workspace) { |
|
|
|
|
|
auto addresses = BuildTaskUtils::GetAddresses(op_desc_, param, keep_workspace); |
|
|
auto all_addresses = BuildTaskUtils::JoinAddresses(addresses); |
|
|
auto all_addresses = BuildTaskUtils::JoinAddresses(addresses); |
|
|
uintptr_t *arg_base = nullptr; |
|
|
uintptr_t *arg_base = nullptr; |
|
|
size_t arg_num = 0; |
|
|
size_t arg_num = 0; |
|
@@ -132,6 +133,10 @@ Status OpTask::UpdateArgTable(const SingleOpModelParam ¶m) { |
|
|
return SUCCESS; |
|
|
return SUCCESS; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
Status OpTask::UpdateArgTable(const SingleOpModelParam ¶m) { |
|
|
|
|
|
return DoUpdateArgTable(param, true); |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
Status OpTask::LaunchKernel(const vector<GeTensorDesc> &input_desc, |
|
|
Status OpTask::LaunchKernel(const vector<GeTensorDesc> &input_desc, |
|
|
const vector<DataBuffer> &input_buffers, |
|
|
const vector<DataBuffer> &input_buffers, |
|
|
vector<GeTensorDesc> &output_desc, |
|
|
vector<GeTensorDesc> &output_desc, |
|
@@ -792,10 +797,9 @@ Status AiCpuTask::LaunchKernel(const std::vector<GeTensorDesc> &input_desc, |
|
|
return SUCCESS; |
|
|
return SUCCESS; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
Status AiCpuTask::UpdateArgTable(const SingleOpModelParam ¶m) { |
|
|
|
|
|
auto addresses = BuildTaskUtils::GetAddresses(op_desc_, param, false); |
|
|
|
|
|
io_addr_host_ = BuildTaskUtils::JoinAddresses(addresses); |
|
|
|
|
|
return SUCCESS; |
|
|
|
|
|
|
|
|
Status AiCpuBaseTask::UpdateArgTable(const SingleOpModelParam ¶m) { |
|
|
|
|
|
// aicpu do not have workspace, for now |
|
|
|
|
|
return DoUpdateArgTable(param, false); |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
void AiCpuTask::GetIoAddr(uintptr_t *&arg_base, size_t &arg_count) { |
|
|
void AiCpuTask::GetIoAddr(uintptr_t *&arg_base, size_t &arg_count) { |
|
|