Browse Source

!1950 Fix bug of tiling workspace.

Merge pull request !1950 from 赵之轩/my_dev
tags/v1.5.1
i-robot Gitee 3 years ago
parent
commit
6ef08b1f33
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      ge/hybrid/node_executor/aicore/aicore_op_task.cc

+ 5
- 3
ge/hybrid/node_executor/aicore/aicore_op_task.cc View File

@@ -372,9 +372,6 @@ Status AiCoreOpTask::UpdateTilingInfo(TaskContext &context) {
// update op args by tiling info
block_dim_ = tiling_info.GetBlockDim();
clear_atomic_ = tiling_info.GetClearAtomic();
std::vector<int64_t> workspaces;
tiling_info.GetAllWorkspaces(workspaces);
op_desc->SetWorkspaceBytes(workspaces);

tiling_data_ = tiling_info.GetAllTilingData().str();
tiling_key_ = tiling_info.GetTilingKey();
@@ -417,6 +414,11 @@ Status AiCoreOpTask::CalcTilingInfo(const NodePtr &node, OpRunInfo &tiling_info)
GE_CHK_STATUS_RET(optiling::OpParaCalculateV2(*node, tiling_info),
"[Invoke][OpParaCalculate]Failed calc tiling data of node %s.",
node->GetName().c_str());
// Only non atomic task need update workspace
auto op_desc = node->GetOpDesc();
std::vector<int64_t> workspaces;
tiling_info.GetAllWorkspaces(workspaces);
op_desc->SetWorkspaceBytes(workspaces);
GELOGD("[%s] Done invoking OpParaCalculate successfully.", node->GetName().c_str());
return SUCCESS;
}


Loading…
Cancel
Save