From e8d0bbb072bfa6cb2ffb4f77562fd8d073194868 Mon Sep 17 00:00:00 2001
From: wuweikang <wuweikang@huawei.com>
Date: Wed, 20 Jan 2021 21:13:46 +0800
Subject: [PATCH] add KernelDefWithHandle

---
 metadef                            |  2 +-
 parser/caffe/proto/task.proto      | 14 ++++++++++++++
 parser/proto/task.proto            | 14 ++++++++++++++
 parser/tensorflow/proto/task.proto | 14 ++++++++++++++
 4 files changed, 43 insertions(+), 1 deletion(-)

diff --git a/metadef b/metadef
index 7f5f6d8..848cf41 160000
--- a/metadef
+++ b/metadef
@@ -1 +1 @@
-Subproject commit 7f5f6d8e22ccff3da85be5920787c095ef0f6e8a
+Subproject commit 848cf412caa9b42ce4e75ab4d0a147ec97dc579b
diff --git a/parser/caffe/proto/task.proto b/parser/caffe/proto/task.proto
index d0c0984..0da5631 100644
--- a/parser/caffe/proto/task.proto
+++ b/parser/caffe/proto/task.proto
@@ -57,6 +57,7 @@ message TaskDef {
     LabelSetDef label_set = 37;
     LabelGotoExDef label_goto_ex = 38;
     LabelSwitchByIndexDef label_switch_by_index = 39;
+    KernelDefWithHandle kernel_with_handle = 40;
 }
 
 message KernelDef {
@@ -74,6 +75,19 @@ message KernelDef {
     uint32 kernel_ext_info_size = 19;
 }
 
+message KernelDefWithHandle {
+    KernelContext context = 1;
+
+    uint64 handle = 10;
+    string dev_func = 11;
+    uint32 block_dim = 12;
+    uint32 args_size = 13;
+    bytes args = 14;
+    bytes sm_desc = 15;
+    string original_kernel_key = 16;
+    string node_info = 17;
+}
+
 message KernelContext {
     uint32 kernel_type = 1;
     uint32 op_id = 2;                              // OP type in CCE
diff --git a/parser/proto/task.proto b/parser/proto/task.proto
index d0c0984..0da5631 100644
--- a/parser/proto/task.proto
+++ b/parser/proto/task.proto
@@ -57,6 +57,7 @@ message TaskDef {
     LabelSetDef label_set = 37;
     LabelGotoExDef label_goto_ex = 38;
     LabelSwitchByIndexDef label_switch_by_index = 39;
+    KernelDefWithHandle kernel_with_handle = 40;
 }
 
 message KernelDef {
@@ -74,6 +75,19 @@ message KernelDef {
     uint32 kernel_ext_info_size = 19;
 }
 
+message KernelDefWithHandle {
+    KernelContext context = 1;
+
+    uint64 handle = 10;
+    string dev_func = 11;
+    uint32 block_dim = 12;
+    uint32 args_size = 13;
+    bytes args = 14;
+    bytes sm_desc = 15;
+    string original_kernel_key = 16;
+    string node_info = 17;
+}
+
 message KernelContext {
     uint32 kernel_type = 1;
     uint32 op_id = 2;                              // OP type in CCE
diff --git a/parser/tensorflow/proto/task.proto b/parser/tensorflow/proto/task.proto
index d0c0984..0da5631 100644
--- a/parser/tensorflow/proto/task.proto
+++ b/parser/tensorflow/proto/task.proto
@@ -57,6 +57,7 @@ message TaskDef {
     LabelSetDef label_set = 37;
     LabelGotoExDef label_goto_ex = 38;
     LabelSwitchByIndexDef label_switch_by_index = 39;
+    KernelDefWithHandle kernel_with_handle = 40;
 }
 
 message KernelDef {
@@ -74,6 +75,19 @@ message KernelDef {
     uint32 kernel_ext_info_size = 19;
 }
 
+message KernelDefWithHandle {
+    KernelContext context = 1;
+
+    uint64 handle = 10;
+    string dev_func = 11;
+    uint32 block_dim = 12;
+    uint32 args_size = 13;
+    bytes args = 14;
+    bytes sm_desc = 15;
+    string original_kernel_key = 16;
+    string node_info = 17;
+}
+
 message KernelContext {
     uint32 kernel_type = 1;
     uint32 op_id = 2;                              // OP type in CCE