taoxiangdong
4 years ago
parent
dc72f5dcae
commit
b9244c3081
@ -1,127 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
package toolkit.dumpdata;
|
||||
|
||||
enum OutputDataType {
|
||||
DT_UNDEFINED = 0;
|
||||
DT_FLOAT = 1;
|
||||
DT_FLOAT16 = 2;
|
||||
DT_INT8 = 3;
|
||||
DT_UINT8 = 4;
|
||||
DT_INT16 = 5;
|
||||
DT_UINT16 = 6;
|
||||
DT_INT32 = 7;
|
||||
DT_INT64 = 8;
|
||||
DT_UINT32 = 9;
|
||||
DT_UINT64 = 10;
|
||||
DT_BOOL = 11;
|
||||
DT_DOUBLE = 12;
|
||||
DT_STRING = 13;
|
||||
DT_DUAL_SUB_INT8 = 14;
|
||||
DT_DUAL_SUB_UINT8 = 15;
|
||||
DT_COMPLEX64 = 16;
|
||||
DT_COMPLEX128 = 17;
|
||||
DT_QINT8 = 18;
|
||||
DT_QINT16 = 19;
|
||||
DT_QINT32 = 20;
|
||||
DT_QUINT8 = 21;
|
||||
DT_QUINT16 = 22;
|
||||
DT_RESOURCE = 23;
|
||||
DT_STRING_REF = 24;
|
||||
DT_DUAL = 25;
|
||||
}
|
||||
|
||||
enum OutputFormat {
|
||||
FORMAT_NCHW = 0;
|
||||
FORMAT_NHWC = 1;
|
||||
FORMAT_ND = 2;
|
||||
FORMAT_NC1HWC0 = 3;
|
||||
FORMAT_FRACTAL_Z = 4;
|
||||
FORMAT_NC1C0HWPAD = 5;
|
||||
FORMAT_NHWC1C0 = 6;
|
||||
FORMAT_FSR_NCHW = 7;
|
||||
FORMAT_FRACTAL_DECONV = 8;
|
||||
FORMAT_C1HWNC0 = 9;
|
||||
FORMAT_FRACTAL_DECONV_TRANSPOSE = 10;
|
||||
FORMAT_FRACTAL_DECONV_SP_STRIDE_TRANS = 11;
|
||||
FORMAT_NC1HWC0_C04 = 12;
|
||||
FORMAT_FRACTAL_Z_C04 = 13;
|
||||
FORMAT_CHWN = 14;
|
||||
FORMAT_FRACTAL_DECONV_SP_STRIDE8_TRANS = 15;
|
||||
FORMAT_HWCN = 16;
|
||||
FORMAT_NC1KHKWHWC0 = 17;
|
||||
FORMAT_BN_WEIGHT = 18;
|
||||
FORMAT_FILTER_HWCK = 19;
|
||||
FORMAT_HASHTABLE_LOOKUP_LOOKUPS=20;
|
||||
FORMAT_HASHTABLE_LOOKUP_KEYS = 21;
|
||||
FORMAT_HASHTABLE_LOOKUP_VALUE = 22;
|
||||
FORMAT_HASHTABLE_LOOKUP_OUTPUT = 23;
|
||||
FORMAT_HASHTABLE_LOOKUP_HITS=24;
|
||||
FORMAT_C1HWNCoC0 = 25;
|
||||
FORMAT_MD = 26;
|
||||
FORMAT_NDHWC = 27;
|
||||
FORMAT_FRACTAL_ZZ = 28;
|
||||
FORMAT_FRACTAL_NZ = 29;
|
||||
FORMAT_RESERVED = 30;
|
||||
}
|
||||
|
||||
message OriginalOp {
|
||||
string name = 1;
|
||||
uint32 output_index = 2;
|
||||
OutputDataType data_type = 3;
|
||||
OutputFormat format = 4;
|
||||
}
|
||||
|
||||
message Shape {
|
||||
repeated uint64 dim = 1;
|
||||
}
|
||||
|
||||
message OpOutput {
|
||||
OutputDataType data_type = 1;
|
||||
OutputFormat format = 2;
|
||||
Shape shape = 3;
|
||||
OriginalOp original_op = 4; // the original op corresponding to the output
|
||||
bytes data = 5;
|
||||
uint64 size = 6;
|
||||
}
|
||||
|
||||
message OpInput {
|
||||
OutputDataType data_type = 1;
|
||||
OutputFormat format = 2;
|
||||
Shape shape = 3;
|
||||
bytes data = 4;
|
||||
uint64 size = 5;
|
||||
}
|
||||
|
||||
enum BufferType {
|
||||
L1 = 0;
|
||||
}
|
||||
|
||||
message OpBuffer {
|
||||
BufferType buffer_type = 1;
|
||||
bytes data = 2;
|
||||
uint64 size = 3;
|
||||
}
|
||||
|
||||
message DumpData{
|
||||
string version = 1;
|
||||
uint64 dump_time = 2;
|
||||
repeated OpOutput output = 3;
|
||||
repeated OpInput input = 4;
|
||||
repeated OpBuffer buffer = 5;
|
||||
}
|
||||
@ -1,26 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
|
||||
import "om.proto";
|
||||
|
||||
package domi;
|
||||
|
||||
message FusionModelDef {
|
||||
string version = 1;
|
||||
repeated OpDef fusion_op = 2;
|
||||
}
|
||||
@ -1,42 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
|
||||
package aicpu.FWKAdapter;
|
||||
option cc_enable_arenas = true;
|
||||
|
||||
|
||||
// Defines an struct for input and output.
|
||||
message TensorDataInfo {
|
||||
|
||||
// value DataType
|
||||
uint32 dtype = 1;
|
||||
|
||||
// shape dim
|
||||
repeated int64 dim = 2;
|
||||
|
||||
// data point addr
|
||||
int64 data_addr = 3;
|
||||
}
|
||||
|
||||
message KernelRunParam {
|
||||
// input
|
||||
repeated TensorDataInfo input = 1;
|
||||
// output
|
||||
repeated TensorDataInfo output = 2;
|
||||
}
|
||||
|
||||
@ -1,104 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
package ge.api_pb;
|
||||
|
||||
import "ge_ir.proto";
|
||||
|
||||
// GE initialize
|
||||
message GEInitialize {
|
||||
map<string, string> options = 1;
|
||||
};
|
||||
|
||||
// initialize response
|
||||
message GEInitializeResponse {
|
||||
uint32 status = 1;
|
||||
uint32 clientId = 2;
|
||||
};
|
||||
|
||||
// GE finalize
|
||||
message GEFinalize {
|
||||
bool final = 1;
|
||||
uint32 clientId = 2;
|
||||
};
|
||||
|
||||
message GEFinalizeResponse {
|
||||
uint32 status = 1;
|
||||
};
|
||||
|
||||
// GE Session
|
||||
message CreateSession{
|
||||
map<string, string> options = 1;
|
||||
};
|
||||
|
||||
message CreateSessionResponse {
|
||||
uint32 status = 1;
|
||||
uint64 sessionId = 2;
|
||||
};
|
||||
|
||||
//GE AddGraph
|
||||
//model serialize :: serializegraph
|
||||
message SessionAddGraph{
|
||||
uint32 graphId = 1;
|
||||
uint64 sessionId = 2;
|
||||
ge.proto.GraphDef graph = 3;
|
||||
};
|
||||
|
||||
message SessionAddGraphResponse {
|
||||
uint32 status = 1;
|
||||
};
|
||||
|
||||
//GE SessionRemoveGraph
|
||||
message SessionRemoveGraph{
|
||||
uint32 graphId = 1;
|
||||
uint64 sessionId = 2;
|
||||
};
|
||||
|
||||
message SessionRemoveGraphResponse {
|
||||
uint32 status = 1;
|
||||
};
|
||||
|
||||
message SessionRunGraph{
|
||||
uint32 graphId = 1;
|
||||
uint64 sessionId = 2;
|
||||
repeated ge.proto.TensorDef tensor = 3;
|
||||
};
|
||||
|
||||
message SessionBuildGraph{
|
||||
uint32 graphId = 1;
|
||||
uint64 sessionId = 2;
|
||||
repeated ge.proto.TensorDef tensor = 3;
|
||||
string savePath = 4;
|
||||
};
|
||||
|
||||
message SessionRunGraphResponse {
|
||||
uint32 status = 1;
|
||||
repeated ge.proto.TensorDef tensor = 2;
|
||||
};
|
||||
|
||||
message SessionBuildGraphResponse {
|
||||
uint32 status = 1;
|
||||
};
|
||||
|
||||
message DestroySession{
|
||||
bool final = 1;
|
||||
uint64 sessionId = 2;
|
||||
};
|
||||
|
||||
message DestroySessionResponse {
|
||||
uint32 status = 1;
|
||||
};
|
||||
File diff suppressed because it is too large
Load Diff
@ -1,89 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
package aicpu.dump;
|
||||
|
||||
message Shape {
|
||||
repeated uint64 dim = 1;
|
||||
}
|
||||
|
||||
message Output {
|
||||
int32 data_type = 1;
|
||||
int32 format = 2;
|
||||
Shape shape = 3;
|
||||
uint64 address = 4;
|
||||
string original_name = 5;
|
||||
int32 original_output_index = 6;
|
||||
int32 original_output_data_type = 7;
|
||||
int32 original_output_format = 8;
|
||||
uint64 size = 9;
|
||||
}
|
||||
|
||||
message Input {
|
||||
int32 data_type =1;
|
||||
int32 format = 2;
|
||||
Shape shape = 3;
|
||||
uint64 address = 4;
|
||||
uint64 size = 5;
|
||||
}
|
||||
|
||||
enum BufferType {
|
||||
L1 = 0;
|
||||
}
|
||||
|
||||
message OpBuffer {
|
||||
BufferType buffer_type = 1;
|
||||
uint64 address = 2;
|
||||
uint64 size = 3;
|
||||
}
|
||||
|
||||
message Op {
|
||||
string op_name = 1;
|
||||
string op_type = 2;
|
||||
}
|
||||
|
||||
message Task {
|
||||
uint32 task_id = 1;
|
||||
uint32 stream_id = 2;
|
||||
Op op = 3;
|
||||
repeated Output output = 4;
|
||||
bool end_graph = 5;
|
||||
repeated Input input = 6;
|
||||
repeated OpBuffer buffer = 7;
|
||||
}
|
||||
|
||||
message OpMappingInfo {
|
||||
string dump_path = 1;
|
||||
oneof model_name_param {
|
||||
string model_name = 2;
|
||||
}
|
||||
oneof model_id_param {
|
||||
uint32 model_id = 3;
|
||||
}
|
||||
oneof step_id {
|
||||
uint64 step_id_addr = 4;
|
||||
}
|
||||
oneof iterations_per_loop {
|
||||
uint64 iterations_per_loop_addr = 5;
|
||||
}
|
||||
oneof loop_cond {
|
||||
uint64 loop_cond_addr = 6;
|
||||
}
|
||||
uint32 flag = 7; // 0x01 load, 0x00 unload
|
||||
repeated Task task = 8;
|
||||
string dump_step = 9;
|
||||
}
|
||||
@ -1,23 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
package ge.optimizers;
|
||||
|
||||
// Default: GE>FE>AICPU
|
||||
message Priority{
|
||||
repeated string optimizer = 1;
|
||||
}
|
||||
@ -1,170 +0,0 @@
|
||||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
syntax = "proto3";
|
||||
|
||||
package domi;
|
||||
|
||||
message ModelTaskDef {
|
||||
string version = 1;
|
||||
|
||||
map<string, string> attr = 9; // Extended field
|
||||
repeated TaskDef task = 10;
|
||||
|
||||
uint64 memory_size = 11;
|
||||
uint32 stream_num = 12;
|
||||
uint32 event_num = 13;
|
||||
uint64 weight_size = 14;
|
||||
|
||||
repeated bytes op = 15; // input/output opdef in bytes
|
||||
|
||||
uint64 base_addr = 16; // base addr
|
||||
uint64 weight_addr = 17; // weight addr
|
||||
uint32 batch_num = 18;
|
||||
}
|
||||
|
||||
|
||||
message TaskDef {
|
||||
uint32 id = 1;
|
||||
uint32 type = 2;
|
||||
|
||||
uint32 stream_id = 10;
|
||||
uint32 event_id = 11;
|
||||
|
||||
KernelDef kernel = 20;
|
||||
KernelExDef kernel_ex = 21;
|
||||
KernelHcclDef kernel_hccl = 25;
|
||||
EventExDef event_ex = 26;
|
||||
LogTimeStampDef log_timestamp = 28;
|
||||
|
||||
uint32 label_id = 30;
|
||||
|
||||
MemcpyAsyncDef memcpy_async = 31;
|
||||
StreamSwitchDef stream_switch = 32;
|
||||
StreamActiveDef stream_active = 33;
|
||||
bytes private_def = 34;
|
||||
uint64 ops_kernel_store_ptr = 35; // adjustments to other fields in the future
|
||||
StreamSwitchNDef stream_switch_n = 36;
|
||||
|
||||
LabelSetDef label_set = 37;
|
||||
LabelGotoExDef label_goto_ex = 38;
|
||||
LabelSwitchByIndexDef label_switch_by_index = 39;
|
||||
}
|
||||
|
||||
message KernelDef {
|
||||
KernelContext context = 1;
|
||||
|
||||
string stub_func = 10;
|
||||
uint32 block_dim = 11;
|
||||
uint32 args_size = 12;
|
||||
bytes args = 13;
|
||||
bytes sm_desc = 14;
|
||||
bytes flowtable = 15;
|
||||
string so_name = 16;
|
||||
string kernel_name = 17;
|
||||
bytes kernel_ext_info = 18;
|
||||
uint32 kernel_ext_info_size = 19;
|
||||
}
|
||||
|
||||
message KernelContext {
|
||||
uint32 kernel_type = 1;
|
||||
uint32 op_id = 2; // OP type in CCE
|
||||
uint32 kernel_func_id = 3;
|
||||
uint32 op_index = 4; // TE/Custom operator
|
||||
bool is_flowtable = 5; // Identify whether args is a flowtable structure
|
||||
bytes args_offset = 6; // args offset information
|
||||
uint32 args_count = 7; // args count
|
||||
repeated uint32 origin_op_index = 8;
|
||||
}
|
||||
|
||||
|
||||
message KernelExDef {
|
||||
uint32 flags = 1;
|
||||
|
||||
uint32 op_index = 4;
|
||||
uint32 args_size = 12;
|
||||
bytes args = 13;
|
||||
bytes task_info = 14; // serialized nodeDef, funcDef, inputoutput
|
||||
uint32 task_info_size = 15;
|
||||
bytes kernel_ext_info = 16;
|
||||
uint32 kernel_ext_info_size = 17;
|
||||
}
|
||||
|
||||
|
||||
message KernelHcclDef {
|
||||
uint32 op_index = 8;
|
||||
string hccl_type = 9;
|
||||
}
|
||||
|
||||
|
||||
message EventExDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 event_type = 2;
|
||||
}
|
||||
|
||||
message LogTimeStampDef {
|
||||
uint64 logid = 1;
|
||||
bool notify = 2;
|
||||
uint32 flat = 3;
|
||||
}
|
||||
|
||||
message MemcpyAsyncDef {
|
||||
uint64 dst = 1;
|
||||
uint64 dst_max = 2;
|
||||
uint64 src = 3;
|
||||
uint64 count = 4;
|
||||
uint32 kind = 5;
|
||||
uint32 op_index = 6;
|
||||
}
|
||||
|
||||
message StreamSwitchDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 true_stream_id = 2;
|
||||
int64 value = 3;
|
||||
uint64 value_ptr = 4;
|
||||
uint32 data_type = 5;
|
||||
}
|
||||
|
||||
message StreamActiveDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 active_stream_id = 2;
|
||||
}
|
||||
|
||||
message StreamSwitchNDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 size = 2;
|
||||
repeated int64 target_value = 3;
|
||||
repeated uint32 true_stream_id = 4;
|
||||
uint32 element_size = 5;
|
||||
uint32 data_type = 6;
|
||||
}
|
||||
|
||||
message LabelSetDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 label_id = 2;
|
||||
uint32 model_id = 3;
|
||||
}
|
||||
|
||||
message LabelGotoExDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 label_id = 2;
|
||||
uint32 model_id = 3;
|
||||
}
|
||||
|
||||
message LabelSwitchByIndexDef {
|
||||
uint32 op_index = 1;
|
||||
uint32 label_max = 2;
|
||||
}
|
||||
Loading…
Reference in new issue