Browse Source

add KernelDefWithHandle

pull/217/head
wuweikang 5 years ago
parent
commit
e8d0bbb072
4 changed files with 43 additions and 1 deletions
  1. +1
    -1
      metadef
  2. +14
    -0
      parser/caffe/proto/task.proto
  3. +14
    -0
      parser/proto/task.proto
  4. +14
    -0
      parser/tensorflow/proto/task.proto

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 7f5f6d8e22ccff3da85be5920787c095ef0f6e8a
Subproject commit 848cf412caa9b42ce4e75ab4d0a147ec97dc579b

+ 14
- 0
parser/caffe/proto/task.proto View File

@@ -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


+ 14
- 0
parser/proto/task.proto View File

@@ -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


+ 14
- 0
parser/tensorflow/proto/task.proto View File

@@ -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


Loading…
Cancel
Save