From 8081a449a09c1d21f7f7ab7589dd408ec9090e49 Mon Sep 17 00:00:00 2001
From: l00444296 <lixiwen1@huawei.com>
Date: Tue, 1 Dec 2020 14:04:07 +0800
Subject: [PATCH] Feature: Support single op profiling

---
 ge/single_op/single_op_model.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/single_op/single_op_model.cc b/ge/single_op/single_op_model.cc
index a0b0d608..2439a82b 100755
--- a/ge/single_op/single_op_model.cc
+++ b/ge/single_op/single_op_model.cc
@@ -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 {