Browse Source

!698 C83 告警清理

Merge pull request !698 from 刘强/lastoneLQ
pull/701/head
刘强 i-robot 2 years ago
parent
commit
e6ab48c47c
10 changed files with 56 additions and 56 deletions
  1. +2
    -2
      parser/caffe/caffe_parser.cc
  2. +24
    -24
      parser/common/acl_graph_parser_util.cc
  3. +3
    -3
      parser/common/acl_graph_parser_util.h
  4. +4
    -4
      parser/onnx/onnx_parser.cc
  5. +2
    -2
      parser/tensorflow/tensorflow_parser.cc
  6. +2
    -2
      tests/st/testcase/test_caffe_parser.cc
  7. +7
    -7
      tests/st/testcase/test_tensorflow_parser.cc
  8. +4
    -4
      tests/ut/parser/testcase/caffe_parser_testcase/caffe_parser_unittest.cc
  9. +1
    -1
      tests/ut/parser/testcase/common/acl_graph_parser_unittest.cc
  10. +7
    -7
      tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc

+ 2
- 2
parser/caffe/caffe_parser.cc View File

@@ -86,7 +86,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file,
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::CAFFE)));

// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
domi::Status status = acl_graph_parse_util.AclParserInitialize(options);
if (status != domi::SUCCESS) {
REPORT_CALL_ERROR("E19999", "AclParserInitialize failed, ret:%d.", status);
@@ -144,7 +144,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file,
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::CAFFE)));

// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
domi::Status status = acl_graph_parse_util.AclParserInitialize(options);
if (status != domi::SUCCESS) {
REPORT_CALL_ERROR("E19999", "AclParserInitialize failed, ret:%d.", status);


+ 24
- 24
parser/common/acl_graph_parser_util.cc View File

@@ -149,7 +149,7 @@ static Status CheckOutNode(ge::OpDescPtr op_desc, int32_t index) {
return domi::SUCCESS;
}

domi::Status AclGraphParseUtil::LoadOpsProtoLib() {
domi::Status AclGraphParserUtil::LoadOpsProtoLib() {
string opsproto_path;
ge::Status ret = ge::TBEPluginLoader::GetOpsProtoPath(opsproto_path);
if (ret != ge::SUCCESS) {
@@ -170,7 +170,7 @@ domi::Status AclGraphParseUtil::LoadOpsProtoLib() {
return SUCCESS;
}

void AclGraphParseUtil::SaveCustomCaffeProtoPath() {
void AclGraphParserUtil::SaveCustomCaffeProtoPath() {
GELOGD("Enter save custom caffe proto path.");
std::string path_base = GetSoPath();
path_base = path_base.substr(0, path_base.rfind('/'));
@@ -192,7 +192,7 @@ void AclGraphParseUtil::SaveCustomCaffeProtoPath() {

// Initialize PARSER, load custom op plugin
// options will be used later for parser decoupling
domi::Status AclGraphParseUtil::AclParserInitialize(const std::map<std::string, std::string> &options) {
domi::Status AclGraphParserUtil::AclParserInitialize(const std::map<std::string, std::string> &options) {
GELOGT(TRACE_INIT, "AclParserInitialize start");
// check init status
if (parser_initialized) {
@@ -240,7 +240,7 @@ domi::Status AclGraphParseUtil::AclParserInitialize(const std::map<std::string,
return SUCCESS;
}

void AclGraphParseUtil::SetDefaultFormat() {
void AclGraphParserUtil::SetDefaultFormat() {
if (ge::GetParserContext().type == domi::TENSORFLOW) {
ge::GetParserContext().format = domi::DOMI_TENSOR_NHWC;
} else {
@@ -248,7 +248,7 @@ void AclGraphParseUtil::SetDefaultFormat() {
}
}

domi::Status AclGraphParseUtil::ParseAclOutputNodes(const string &out_nodes) const {
domi::Status AclGraphParserUtil::ParseAclOutputNodes(const string &out_nodes) const {
try {
ge::GetParserContext().out_nodes_map.clear();
ge::GetParserContext().user_out_nodes.clear();
@@ -323,7 +323,7 @@ domi::Status AclGraphParseUtil::ParseAclOutputNodes(const string &out_nodes) con
return SUCCESS;
}

domi::Status AclGraphParseUtil::ParseAclOutputFp16NodesFormat(const string &is_output_fp16) const {
domi::Status AclGraphParserUtil::ParseAclOutputFp16NodesFormat(const string &is_output_fp16) const {
if (is_output_fp16.empty()) {
return SUCCESS;
}
@@ -347,7 +347,7 @@ domi::Status AclGraphParseUtil::ParseAclOutputFp16NodesFormat(const string &is_o
return SUCCESS;
}

domi::Status AclGraphParseUtil::ParseAclEnableScope(const string &enable_scope_fusion_passes) const {
domi::Status AclGraphParserUtil::ParseAclEnableScope(const string &enable_scope_fusion_passes) const {
ge::GetParserContext().enable_scope_fusion_passes.clear();
if (enable_scope_fusion_passes.empty()) {
return SUCCESS;
@@ -356,8 +356,8 @@ domi::Status AclGraphParseUtil::ParseAclEnableScope(const string &enable_scope_f
return SUCCESS;
}

void AclGraphParseUtil::AddAttrsForInputNodes(const vector<string> &adjust_fp16_format_vec,
const string &fp16_nodes_name, size_t index, OpDescPtr &op_desc) {
void AclGraphParserUtil::AddAttrsForInputNodes(const vector<string> &adjust_fp16_format_vec,
const string &fp16_nodes_name, size_t index, OpDescPtr &op_desc) {
if (AttrUtils::SetStr(op_desc, ATTR_ATC_USER_DEFINE_DATATYPE, TypeUtils::DataTypeToSerialString(DT_FLOAT16))) {
if ((index < adjust_fp16_format_vec.size()) && (adjust_fp16_format_vec[index] == "true")) {
GELOGI("This node [%s] should be set NC1HWC0", fp16_nodes_name.c_str());
@@ -368,8 +368,8 @@ void AclGraphParseUtil::AddAttrsForInputNodes(const vector<string> &adjust_fp16_
}
}

domi::Status AclGraphParseUtil::ParseAclInputFp16Nodes(const ComputeGraphPtr &graph, const string &input_fp16_nodes,
const string &is_input_adjust_hw_layout) const {
domi::Status AclGraphParserUtil::ParseAclInputFp16Nodes(const ComputeGraphPtr &graph, const string &input_fp16_nodes,
const string &is_input_adjust_hw_layout) const {
GE_CHECK_NOTNULL(graph);
vector<string> adjust_fp16_format_vec;
if (!is_input_adjust_hw_layout.empty()) {
@@ -411,7 +411,7 @@ domi::Status AclGraphParseUtil::ParseAclInputFp16Nodes(const ComputeGraphPtr &gr
return SUCCESS;
}

domi::Status AclGraphParseUtil::SetSpecifyIndexAttrByInputNames(const ComputeGraphPtr &graph,
domi::Status AclGraphParserUtil::SetSpecifyIndexAttrByInputNames(const ComputeGraphPtr &graph,
const std::string &input_data_names) const {
std::vector<std::string> input_names = StringUtils::Split(input_data_names, ',');
std::unordered_map<std::string, size_t> name_to_index;
@@ -446,8 +446,8 @@ domi::Status AclGraphParseUtil::SetSpecifyIndexAttrByInputNames(const ComputeGra
return SUCCESS;
}

void AclGraphParseUtil::CreateOutputNodesInfo(std::vector<std::pair<ge::NodePtr, int32_t>> &output_nodes_info,
std::vector<std::string> &output_nodes_name) const {
void AclGraphParserUtil::CreateOutputNodesInfo(std::vector<std::pair<ge::NodePtr, int32_t>> &output_nodes_info,
std::vector<std::string> &output_nodes_name) const {
output_nodes_name.clear();
auto &out_tensor_names = ge::GetParserContext().out_tensor_names;
if (out_tensor_names.empty()) {
@@ -478,8 +478,8 @@ void AclGraphParseUtil::CreateOutputNodesInfo(std::vector<std::pair<ge::NodePtr,
}
}

domi::Status AclGraphParseUtil::GetOutputLeaf(NodePtr node,
std::vector<std::pair<ge::NodePtr, int32_t>> &output_nodes_info) const {
domi::Status AclGraphParserUtil::GetOutputLeaf(NodePtr node,
std::vector<std::pair<ge::NodePtr, int32_t>> &output_nodes_info) const {
ge::OpDescPtr tmpDescPtr = node->GetOpDesc();
if (tmpDescPtr == nullptr) {
REPORT_INNER_ERROR("E19999", "param node has no opdesc.");
@@ -508,7 +508,7 @@ domi::Status AclGraphParseUtil::GetOutputLeaf(NodePtr node,
return SUCCESS;
}

domi::Status AclGraphParseUtil::GetDefaultOutInfo(ge::ComputeGraphPtr &compute_graph,
domi::Status AclGraphParserUtil::GetDefaultOutInfo(ge::ComputeGraphPtr &compute_graph,
std::vector<std::pair<ge::NodePtr, int32_t>> &output_nodes_info) const {
std::vector<std::pair<std::string, int32_t>> default_out_nodes = ge::GetParserContext().default_out_nodes;
if (!default_out_nodes.empty()) {
@@ -531,8 +531,8 @@ domi::Status AclGraphParseUtil::GetDefaultOutInfo(ge::ComputeGraphPtr &compute_g
return domi::SUCCESS;
}

domi::Status AclGraphParseUtil::SetOutputNodeInfo(ge::Graph &graph,
const std::map<AscendString, AscendString> &parser_params) const {
domi::Status AclGraphParserUtil::SetOutputNodeInfo(ge::Graph &graph,
const std::map<AscendString, AscendString> &parser_params) const {
(void)parser_params;
ge::ComputeGraphPtr compute_graph = ge::GraphUtils::GetComputeGraph(graph);
GE_CHECK_NOTNULL(compute_graph);
@@ -588,7 +588,7 @@ domi::Status AclGraphParseUtil::SetOutputNodeInfo(ge::Graph &graph,
return domi::SUCCESS;
}

domi::Status AclGraphParseUtil::CheckOptions(const std::map<AscendString, AscendString> &parser_params) const {
domi::Status AclGraphParserUtil::CheckOptions(const std::map<AscendString, AscendString> &parser_params) const {
for (auto &ele : parser_params) {
const char *key_ascend = ele.first.GetString();
if (key_ascend == nullptr) {
@@ -609,8 +609,8 @@ domi::Status AclGraphParseUtil::CheckOptions(const std::map<AscendString, Ascend
return SUCCESS;
}

domi::Status AclGraphParseUtil::ParseParamsBeforeGraph(const std::map<AscendString, AscendString> &parser_params,
string &graph_name) const {
domi::Status AclGraphParserUtil::ParseParamsBeforeGraph(const std::map<AscendString, AscendString> &parser_params,
string &graph_name) const {
GELOGI("Parse graph user options start.");
ge::GetParserContext().input_nodes_format_map.clear();
ge::GetParserContext().output_formats.clear();
@@ -663,8 +663,8 @@ domi::Status AclGraphParseUtil::ParseParamsBeforeGraph(const std::map<AscendStri
return SUCCESS;
}

domi::Status AclGraphParseUtil::ParseParamsAfterGraph(ge::Graph &graph,
const std::map<AscendString, AscendString> &parser_params) const {
domi::Status AclGraphParserUtil::ParseParamsAfterGraph(ge::Graph &graph,
const std::map<AscendString, AscendString> &parser_params) const {
// support paragrams: input_fp16_nodes, is_input_adjust_hw_layout,
ComputeGraphPtr compute_graph = GraphUtils::GetComputeGraph(graph);
GE_CHECK_NOTNULL(compute_graph);


+ 3
- 3
parser/common/acl_graph_parser_util.h View File

@@ -34,10 +34,10 @@ namespace ge {

using google::protobuf::Message;

class AclGraphParseUtil {
class AclGraphParserUtil {
public:
AclGraphParseUtil() {}
virtual ~AclGraphParseUtil() {}
AclGraphParserUtil() {}
virtual ~AclGraphParserUtil() {}
static domi::Status LoadOpsProtoLib();
static void SaveCustomCaffeProtoPath();
domi::Status AclParserInitialize(const std::map<std::string, std::string> &options);


+ 4
- 4
parser/onnx/onnx_parser.cc View File

@@ -52,7 +52,7 @@ const char *kLocation = "location";
}

namespace ge {
graphStatus PrepareBeforeParse(AclGraphParseUtil &acl_graph_parse_util,
graphStatus PrepareBeforeParse(AclGraphParserUtil &acl_graph_parse_util,
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph, std::shared_ptr<domi::ModelParser> &model_parser) {
GetParserContext().type = domi::ONNX;
@@ -82,7 +82,7 @@ graphStatus PrepareBeforeParse(AclGraphParseUtil &acl_graph_parse_util,
return ge::SUCCESS;
}

graphStatus HandleAfterParse(AclGraphParseUtil &acl_graph_parse_util,
graphStatus HandleAfterParse(AclGraphParserUtil &acl_graph_parse_util,
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph) {
if (acl_graph_parse_util.ParseParamsAfterGraph(graph, parser_params) != ge::SUCCESS) {
@@ -104,7 +104,7 @@ graphStatus aclgrphParseONNX(const char *model_file,
const std::map<AscendString, AscendString> &parser_params, ge::Graph &graph) {
GE_CHECK_NOTNULL(model_file);
// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::shared_ptr<domi::ModelParser> model_parser;

if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) {
@@ -136,7 +136,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size,
const std::map<AscendString, AscendString> &parser_params, ge::Graph &graph) {
GE_CHECK_NOTNULL(buffer);
// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::shared_ptr<domi::ModelParser> model_parser;

if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) {


+ 2
- 2
parser/tensorflow/tensorflow_parser.cc View File

@@ -94,7 +94,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph) {
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::TENSORFLOW)));

// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
if (acl_graph_parse_util.AclParserInitialize(options) != domi::SUCCESS) {
GELOGE(GRAPH_FAILED, "Parser Initialize failed.");
return GRAPH_FAILED;
@@ -142,7 +142,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, const std::map<Ascend
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::TENSORFLOW)));

// load custom plugin so and proto
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
domi::Status status = acl_graph_parse_util.AclParserInitialize(options);
if (status != domi::SUCCESS) {
GELOGE(GRAPH_FAILED, "Parser Initialize failed.");


+ 2
- 2
tests/st/testcase/test_caffe_parser.cc View File

@@ -191,7 +191,7 @@ TEST_F(STestCaffeParser, caffe_parser_user_output_with_default) {
ge::Graph graph = ge::GraphUtils::CreateGraphFromComputeGraph(compute_graph);
auto ret = model_parser->Parse(model_file.c_str(), graph);
ASSERT_EQ(ret, GRAPH_SUCCESS);
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> parser_params;
auto status = acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params);
ASSERT_EQ(status, SUCCESS);
@@ -483,7 +483,7 @@ TEST_F(STestCaffeParser, CaffeWeightsParser_CreateCustomOperator_test)
TEST_F(STestCaffeParser, CaffeWeightsParser_ParseOutputNodeTopInfo_test)
{
CaffeModelParser model_parser;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;

domi::caffe::NetParameter net;
domi::caffe::LayerParameter *lay0 = net.add_layer();


+ 7
- 7
tests/st/testcase/test_tensorflow_parser.cc View File

@@ -1104,7 +1104,7 @@ TEST_F(STestTensorflowParser, parser_tensorflow_model) {

// parser tensorflow model out_node_size is equal to index
string graph_name;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> out_nodes_with_node_and_index = {
{AscendString(ge::ir_option::OUT_NODES), AscendString("Placeholder:0;Placeholder_1:1")}};
ParerSTestsUtils::ClearParserInnerCtx();
@@ -1356,7 +1356,7 @@ TEST_F(STestTensorflowParser, tensorflow_parserAllGraph_failed)

TEST_F(STestTensorflowParser, test_parse_acl_output_nodes)
{
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
string graph_name;
// case 1: Normal with 'node and index'
ParerSTestsUtils::ClearParserInnerCtx();
@@ -1523,7 +1523,7 @@ TEST_F(STestTensorflowParser, parse_AddFmkNode)
std::string modelFile = caseDir + "/origin_models/tf_add.pb";
ge::Graph graph;
string graph_name;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<ge::AscendString, ge::AscendString> parser_options = {{AscendString(ge::ir_option::OUT_NODES), AscendString("Placeholder:0;Placeholder_1:0")}};
ParerSTestsUtils::ClearParserInnerCtx();
Status ret = acl_graph_parse_util.ParseParamsBeforeGraph(parser_options, graph_name);
@@ -3781,9 +3781,9 @@ TEST_F(STestTensorflowParser, tensorflow_ReadBytesFromBinaryFile_test)
EXPECT_EQ(realPath, "");
}

TEST_F(STestTensorflowParser, tensorflow_AclGraphParseUtil_ParseAclInputFp16Nodes_test)
TEST_F(STestTensorflowParser, tensorflow_AclGraphParserUtil_ParseAclInputFp16Nodes_test)
{
AclGraphParseUtil parserUtil;
AclGraphParserUtil parserUtil;
ge::ComputeGraphPtr graph = std::make_shared<ge::ComputeGraph>(GRAPH_DEFAULT_NAME);
std::string input_fp16_nodes = "Add";
std::string is_input_adjust_hw_layout = "is_input_adjust_hw_layout";
@@ -4010,7 +4010,7 @@ TEST_F(STestTensorflowParser, tensorflow_FP16_parser_test)

TEST_F(STestTensorflowParser, tensorflow_AclParserInitialize_test)
{
AclGraphParseUtil parseUtil;
AclGraphParserUtil parseUtil;
std::map<std::string, std::string> options;
Status ret = parseUtil.AclParserInitialize(options);
EXPECT_EQ(ret, FAILED);
@@ -4022,7 +4022,7 @@ TEST_F(STestTensorflowParser, tensorflow_AclParserInitialize_test)

TEST_F(STestTensorflowParser, tensorflow_GetOutputLeaf_test)
{
AclGraphParseUtil parseUtil;
AclGraphParserUtil parseUtil;
ge::ComputeGraphPtr compute_graph = build_graph(true);
ge::NodePtr output_nodes_info = compute_graph->FindNode("Relu3");
std::vector<std::pair<ge::NodePtr, int32_t>> output_nodes = {{output_nodes_info,0}};


+ 4
- 4
tests/ut/parser/testcase/caffe_parser_testcase/caffe_parser_unittest.cc View File

@@ -189,7 +189,7 @@ TEST_F(UtestCaffeParser, caffe_parser_user_output_with_name_and_index) {
ge::GetParserContext().user_out_nodes.push_back({"abs", 0});
auto ret = model_parser->Parse(model_file.c_str(), graph);
ASSERT_EQ(ret, GRAPH_SUCCESS);
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> parser_params;
auto status = acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params);
ASSERT_EQ(status, SUCCESS);
@@ -216,7 +216,7 @@ TEST_F(UtestCaffeParser, caffe_parser_user_output_with_top_name) {
ge::GetParserContext().user_out_tensors.push_back("abs_out");
auto ret = model_parser->Parse(model_file.c_str(), graph);
ASSERT_EQ(ret, GRAPH_SUCCESS);
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> parser_params;
auto status = acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params);
ASSERT_EQ(status, SUCCESS);
@@ -241,7 +241,7 @@ TEST_F(UtestCaffeParser, caffe_parser_user_output_with_default) {
ge::Graph graph = ge::GraphUtils::CreateGraphFromComputeGraph(compute_graph);
auto ret = model_parser->Parse(model_file.c_str(), graph);
ASSERT_EQ(ret, GRAPH_SUCCESS);
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> parser_params;
auto status = acl_graph_parse_util.SetOutputNodeInfo(graph, parser_params);
ASSERT_EQ(status, SUCCESS);
@@ -543,7 +543,7 @@ TEST_F(UtestCaffeParser, CaffeWeightsParser_CreateCustomOperator_test)
TEST_F(UtestCaffeParser, CaffeWeightsParser_ParseOutputNodeTopInfo_test)
{
CaffeModelParser model_parser;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;

domi::caffe::NetParameter net;
domi::caffe::LayerParameter *lay0 = net.add_layer();


+ 1
- 1
tests/ut/parser/testcase/common/acl_graph_parser_unittest.cc View File

@@ -53,7 +53,7 @@ class UtestAclGraphParser : public testing::Test {
};

TEST_F(UtestAclGraphParser, test_parse_acl_output_nodes) {
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
string graph_name;
// case 1: Normal with 'node and index'
ParerUTestsUtils::ClearParserInnerCtx();


+ 7
- 7
tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc View File

@@ -1106,7 +1106,7 @@ TEST_F(UtestTensorflowParser, parser_tensorflow_model) {

// parser tensorflow model out_node_size is equal to index
string graph_name;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<AscendString, AscendString> out_nodes_with_node_and_index = {
{AscendString(ge::ir_option::OUT_NODES), AscendString("Placeholder:0;Placeholder_1:1")}};
ParerUTestsUtils::ClearParserInnerCtx();
@@ -1452,7 +1452,7 @@ TEST_F(UtestTensorflowParser, tensorflow_parserAllGraph_failed)

TEST_F(UtestTensorflowParser, test_parse_acl_output_nodes)
{
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
string graph_name;
// case 1: Normal with 'node and index'
ParerUTestsUtils::ClearParserInnerCtx();
@@ -1621,7 +1621,7 @@ TEST_F(UtestTensorflowParser, parse_AddFmkNode)
std::string modelFile = caseDir + "/tensorflow_model/tf_add.pb";
ge::Graph graph;
string graph_name;
AclGraphParseUtil acl_graph_parse_util;
AclGraphParserUtil acl_graph_parse_util;
std::map<ge::AscendString, ge::AscendString> parser_options = {{AscendString(ge::ir_option::OUT_NODES), AscendString("Placeholder:0;Placeholder_1:0")}};
ParerUTestsUtils::ClearParserInnerCtx();
Status ret = acl_graph_parse_util.ParseParamsBeforeGraph(parser_options, graph_name);
@@ -3885,9 +3885,9 @@ TEST_F(UtestTensorflowParser, tensorflow_ReadBytesFromBinaryFile_test)
EXPECT_EQ(realPath, "");
}

TEST_F(UtestTensorflowParser, tensorflow_AclGraphParseUtil_ParseAclInputFp16Nodes_test)
TEST_F(UtestTensorflowParser, tensorflow_AclGraphParserUtil_ParseAclInputFp16Nodes_test)
{
AclGraphParseUtil parserUtil;
AclGraphParserUtil parserUtil;
ge::ComputeGraphPtr graph = std::make_shared<ge::ComputeGraph>(GRAPH_DEFAULT_NAME);
std::string input_fp16_nodes = "Add";
std::string is_input_adjust_hw_layout = "is_input_adjust_hw_layout";
@@ -4094,7 +4094,7 @@ TEST_F(UtestTensorflowParser, tensorflow_FP16_parser_test)

TEST_F(UtestTensorflowParser, tensorflow_AclParserInitialize_test)
{
AclGraphParseUtil parseUtil;
AclGraphParserUtil parseUtil;
std::map<std::string, std::string> options;
Status ret = parseUtil.AclParserInitialize(options);
EXPECT_EQ(ret, FAILED);
@@ -4106,7 +4106,7 @@ TEST_F(UtestTensorflowParser, tensorflow_AclParserInitialize_test)

TEST_F(UtestTensorflowParser, tensorflow_GetOutputLeaf_test)
{
AclGraphParseUtil parseUtil;
AclGraphParserUtil parseUtil;
ge::ComputeGraphPtr compute_graph = build_graph(true);
ge::NodePtr output_nodes_info = compute_graph->FindNode("Relu3");
std::vector<std::pair<ge::NodePtr, int32_t>> output_nodes = {{output_nodes_info,0}};


Loading…
Cancel
Save