m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!983 add KernelDefWithHandle

Browse Source 
Merge pull request !983 from HW_KK/task-proto
pull/983/MERGE
计晨 4 years ago committed by Gitee
parent 715bd8eedc b86236fe30
commit 435795139f
9 changed files with 100 additions and 2 deletions
Show Stats Download Patch File Download Diff File

14
ge/client/proto/task.proto
Unescape 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
ge/common/proto/task.proto
Unescape 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
ge/executor/proto/task.proto
Unescape 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
ge/ge_local_engine/proto/task.proto
Unescape 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
ge/host_cpu_engine/proto/task.proto
Unescape 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
ge/offline/proto/task.proto
Unescape 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
ge/proto/task.proto
Unescape 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

2
metadef

@ -1 +1 @@
Subproject commit 88d053a5f94c40ff21620cef50b87075d5054292
Subproject commit 848cf412caa9b42ce4e75ab4d0a147ec97dc579b

2
parser

@ -1 +1 @@
Subproject commit 6904ba9488658afc30076d299183fc8875045f49
Subproject commit 756c64c59e451a955e81b59d957ad55f96c27d89
Write Preview
Loading…
Cancel
Save