taoxiangdong
5 years ago
parent
2ed6c25598
commit
b264e53300
@ -1,163 +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.
|
||||
*/
|
||||
|
||||
#ifndef INC_EXTERNAL_REGISTER_REGISTER_H_
|
||||
#define INC_EXTERNAL_REGISTER_REGISTER_H_
|
||||
|
||||
#include <functional>
|
||||
#include <initializer_list>
|
||||
#include <map>
|
||||
#include <memory>
|
||||
#include <set>
|
||||
#include <string>
|
||||
#include <utility>
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include "graph/operator.h"
|
||||
#include "register/register_error_codes.h"
|
||||
#include "register/register_fmk_types.h"
|
||||
#include "register/register_types.h"
|
||||
|
||||
using std::make_shared;
|
||||
using std::map;
|
||||
using std::pair;
|
||||
using std::string;
|
||||
using std::to_string;
|
||||
using std::unique_ptr;
|
||||
using std::vector;
|
||||
|
||||
/*lint -e148*/
|
||||
namespace ge {
|
||||
class Operator;
|
||||
class TensorDesc;
|
||||
class Tensor;
|
||||
class TBEPluginManager;
|
||||
} // namespace ge
|
||||
|
||||
namespace google {
|
||||
namespace protobuf {
|
||||
class Message;
|
||||
}
|
||||
} // namespace google
|
||||
|
||||
namespace domi {
|
||||
const int64_t kMaxNameLength = 1048576; // 1M
|
||||
|
||||
enum DynamicType { kInvalid = 0, kInput = 1, kOutput = 2 };
|
||||
struct DynamicInputOutputInfo {
|
||||
DynamicType type; // input/output
|
||||
const char *port_name;
|
||||
int64_t port_name_len;
|
||||
const char *attr_name;
|
||||
int64_t attr_name_len;
|
||||
DynamicInputOutputInfo()
|
||||
: type(kInvalid), port_name(nullptr), port_name_len(0), attr_name(nullptr), attr_name_len(0) {}
|
||||
DynamicInputOutputInfo(DynamicType type, const char *port_name, int64_t port_name_len, const char *attr_name,
|
||||
int64_t attr_name_len)
|
||||
: type(type),
|
||||
port_name(port_name),
|
||||
port_name_len(port_name_len),
|
||||
attr_name(attr_name),
|
||||
attr_name_len(attr_name_len) {}
|
||||
};
|
||||
Status AutoMappingByOpFn(const ge::Operator &op_src, ge::Operator &op);
|
||||
Status AutoMappingByOpFnDynamic(const ge::Operator &op_src, ge::Operator &op,
|
||||
const vector<DynamicInputOutputInfo> &dynamic_name_attr_value);
|
||||
Status AutoMappingFn(const google::protobuf::Message *op_src, ge::Operator &op);
|
||||
Status AutoMappingFnDynamic(const google::protobuf::Message *op_src, ge::Operator &op,
|
||||
std::map<std::string, std::pair<std::string, std::string>> dynamic_name_attr_value,
|
||||
int in_pos = -1, int out_pos = -1);
|
||||
Status AutoMappingSubgraphIndex(const ge::Graph &graph, const std::function<int(int data_index)> &input,
|
||||
const std::function<int(int netoutput_index)> &output);
|
||||
Status AutoMappingSubgraphIndex(const ge::Graph &graph,
|
||||
const std::function<Status(int data_index, int &parent_input_index)> &input,
|
||||
const std::function<Status(int netoutput_index, int &parent_output_index)> &output);
|
||||
using google::protobuf::Message;
|
||||
class OpRegistrationDataImpl;
|
||||
|
||||
using ParseParamFunc = std::function<domi::Status(const google::protobuf::Message *, ge::Operator &)>;
|
||||
using ParseParamByOpFunc = std::function<domi::Status(const ge::Operator &, ge::Operator &)>;
|
||||
using FusionParseParamFunc =
|
||||
std::function<domi::Status(const std::vector<const google::protobuf::Message *>, ge::Operator &)>;
|
||||
using FusionParseParamByOpFunc = std::function<domi::Status(const std::vector<ge::Operator> &, ge::Operator &)>;
|
||||
using ParseSubgraphFunc = std::function<Status(const std::string &subgraph_name, const ge::Graph &graph)>;
|
||||
|
||||
class FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY OpRegistrationData {
|
||||
public:
|
||||
OpRegistrationData(const std::string &om_optype);
|
||||
|
||||
~OpRegistrationData();
|
||||
|
||||
OpRegistrationData &FrameworkType(const domi::FrameworkType &fmk_type);
|
||||
|
||||
OpRegistrationData &OriginOpType(const std::initializer_list<std::string> &ori_optype_list);
|
||||
|
||||
OpRegistrationData &OriginOpType(const std::string &ori_optype);
|
||||
|
||||
OpRegistrationData &ParseParamsFn(const ParseParamFunc &parseParamFn);
|
||||
|
||||
OpRegistrationData &ParseParamsByOperatorFn(const ParseParamByOpFunc &parse_param_by_op_fn);
|
||||
|
||||
OpRegistrationData &FusionParseParamsFn(const FusionParseParamFunc &fusionParseParamFn);
|
||||
|
||||
OpRegistrationData &FusionParseParamsFn(const FusionParseParamByOpFunc &fusion_parse_param_fn);
|
||||
|
||||
OpRegistrationData &ParseSubgraphPostFn(const ParseSubgraphFunc &subgraph_post_fn);
|
||||
|
||||
OpRegistrationData &ImplyType(const domi::ImplyType &imply_type);
|
||||
|
||||
OpRegistrationData &DelInputWithCond(int inputIdx, const std::string &attrName, bool attrValue);
|
||||
|
||||
OpRegistrationData &DelInputWithOriginalType(int input_idx, const std::string &ori_type);
|
||||
|
||||
OpRegistrationData &InputReorderVector(const vector<int> &input_order);
|
||||
|
||||
domi::ImplyType GetImplyType() const;
|
||||
std::string GetOmOptype() const;
|
||||
std::set<std::string> GetOriginOpTypeSet() const;
|
||||
domi::FrameworkType GetFrameworkType() const;
|
||||
ParseParamFunc GetParseParamFn() const;
|
||||
ParseParamByOpFunc GetParseParamByOperatorFn() const;
|
||||
FusionParseParamFunc GetFusionParseParamFn() const;
|
||||
FusionParseParamByOpFunc GetFusionParseParamByOpFn() const;
|
||||
ParseSubgraphFunc GetParseSubgraphPostFn() const;
|
||||
|
||||
private:
|
||||
std::shared_ptr<OpRegistrationDataImpl> impl_;
|
||||
friend class OpRegistry;
|
||||
friend class OpRegistrationTbe;
|
||||
friend class ge::TBEPluginManager;
|
||||
};
|
||||
|
||||
class FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY OpReceiver {
|
||||
public:
|
||||
OpReceiver(OpRegistrationData ®_data);
|
||||
~OpReceiver() {}
|
||||
};
|
||||
|
||||
#define REGISTER_CUSTOM_OP(name) REGISTER_CUSTOM_OP_UNIQ_HELPER(__COUNTER__, name)
|
||||
#define REGISTER_CUSTOM_OP_UNIQ_HELPER(ctr, name) REGISTER_CUSTOM_OP_UNIQ(ctr, name)
|
||||
#define REGISTER_CUSTOM_OP_UNIQ(ctr, name) \
|
||||
static OpReceiver register_op##ctr __attribute__((unused)) = OpRegistrationData(name)
|
||||
} // namespace domi
|
||||
|
||||
namespace ge {
|
||||
using OpRegistrationData = domi::OpRegistrationData;
|
||||
using OpReceiver = domi::OpReceiver;
|
||||
} // namespace ge
|
||||
/*lint +e148*/
|
||||
#endif // INC_EXTERNAL_REGISTER_REGISTER_H_
|
||||
@ -1,39 +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.
|
||||
*/
|
||||
|
||||
#ifndef INC_EXTERNAL_REGISTER_REGISTER_ERROR_CODES_H_
|
||||
#define INC_EXTERNAL_REGISTER_REGISTER_ERROR_CODES_H_
|
||||
|
||||
#define SYSID_FWK 3 // Subsystem ID
|
||||
#define MODID_COMMON 0 // Common module ID
|
||||
|
||||
#define DECLARE_ERRORNO(sysid, modid, name, value) \
|
||||
const domi::Status name = \
|
||||
((0xFF & ((uint8_t)sysid)) << 24) | ((0xFF & ((uint8_t)modid)) << 16) | (0xFFFF & ((uint16_t)value));
|
||||
|
||||
#define DECLARE_ERRORNO_COMMON(name, value) DECLARE_ERRORNO(SYSID_FWK, MODID_COMMON, name, value)
|
||||
|
||||
namespace domi {
|
||||
using Status = uint32_t;
|
||||
|
||||
// General error code
|
||||
DECLARE_ERRORNO(0, 0, SUCCESS, 0);
|
||||
DECLARE_ERRORNO(0xFF, 0xFF, FAILED, 0xFFFFFFFF);
|
||||
DECLARE_ERRORNO_COMMON(PARAM_INVALID, 1); // 50331649
|
||||
DECLARE_ERRORNO(SYSID_FWK, 1, SCOPE_NOT_CHANGED, 201);
|
||||
} // namespace domi
|
||||
|
||||
#endif // INC_EXTERNAL_REGISTER_REGISTER_ERROR_CODES_H_
|
||||
@ -1,37 +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.
|
||||
*/
|
||||
|
||||
#ifndef INC_EXTERNAL_REGISTER_REGISTER_FMK_TYPES_H_
|
||||
#define INC_EXTERNAL_REGISTER_REGISTER_FMK_TYPES_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
namespace domi {
|
||||
///
|
||||
/// @ingroup domi_omg
|
||||
/// @brief AI framework types
|
||||
///
|
||||
enum FrameworkType {
|
||||
CAFFE = 0,
|
||||
MINDSPORE = 1,
|
||||
TENSORFLOW = 3,
|
||||
ANDROID_NN,
|
||||
ONNX,
|
||||
FRAMEWORK_RESERVED,
|
||||
};
|
||||
} // namespace domi
|
||||
|
||||
#endif // INC_EXTERNAL_REGISTER_REGISTER_FMK_TYPES_H_
|
||||
@ -1,59 +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.
|
||||
*/
|
||||
|
||||
#ifndef INC_EXTERNAL_REGISTER_REGISTER_TYPES_H_
|
||||
#define INC_EXTERNAL_REGISTER_REGISTER_TYPES_H_
|
||||
|
||||
namespace domi {
|
||||
#ifdef HOST_VISIBILITY
|
||||
#define FMK_FUNC_HOST_VISIBILITY __attribute__((visibility("default")))
|
||||
#else
|
||||
#define FMK_FUNC_HOST_VISIBILITY
|
||||
#endif
|
||||
#ifdef DEV_VISIBILITY
|
||||
#define FMK_FUNC_DEV_VISIBILITY __attribute__((visibility("default")))
|
||||
#else
|
||||
#define FMK_FUNC_DEV_VISIBILITY
|
||||
#endif
|
||||
|
||||
/// CCE defined constant
|
||||
|
||||
///
|
||||
/// @ingroup domi
|
||||
/// @brief original tensor type
|
||||
///
|
||||
typedef enum tagDomiTensorFormat {
|
||||
DOMI_TENSOR_NCHW = 0, // < NCHW
|
||||
DOMI_TENSOR_NHWC, // < NHWC
|
||||
DOMI_TENSOR_ND, // < Nd Tensor
|
||||
DOMI_TENSOR_NC1HWC0, // < NC1HWC0
|
||||
DOMI_TENSOR_FRACTAL_Z, // < FRACTAL_Z
|
||||
DOMI_TENSOR_NC1C0HWPAD,
|
||||
DOMI_TENSOR_NHWC1C0,
|
||||
DOMI_TENSOR_FSR_NCHW,
|
||||
DOMI_TENSOR_FRACTAL_DECONV,
|
||||
DOMI_TENSOR_BN_WEIGHT,
|
||||
DOMI_TENSOR_CHWN, // Android NN Depth CONV
|
||||
DOMI_TENSOR_FILTER_HWCK, // filter input tensor format
|
||||
DOMI_TENSOR_NDHWC,
|
||||
DOMI_TENSOR_NCDHW,
|
||||
DOMI_TENSOR_DHWCN, // 3D filter input tensor format
|
||||
DOMI_TENSOR_DHWNC,
|
||||
DOMI_TENSOR_RESERVED
|
||||
} domiTensorFormat_t;
|
||||
} // namespace domi
|
||||
|
||||
#endif // INC_EXTERNAL_REGISTER_REGISTER_TYPES_H_
|
||||
File diff suppressed because it is too large
Load Diff
@ -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