Browse Source

Feature: Support single op profiling

pull/473/head
l00444296 4 years ago
parent
commit
8081a449a0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/single_op_model.cc

+ 1
- 1
ge/single_op/single_op_model.cc View File

@@ -279,7 +279,7 @@ Status SingleOpModel::BuildTaskList(SingleOp &single_op) {
if (ret != SUCCESS) {
return ret;
}
string op_name = op_list_[context.op_index()]->GetName();
string op_name = op_list_[task_def.kernel_ex().op_index()]->GetName();
single_op.tasks_.emplace_back(aicpu_task);
single_op.op_name_.emplace_back(op_name);
} else {


Loading…
Cancel
Save