Browse Source

fix repeated compute graph bug

pull/568/head
“jiangrun” 3 years ago
parent
commit
8a0f2a71e0
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      parser/common/acl_graph_parser_util.cc

+ 4
- 1
parser/common/acl_graph_parser_util.cc View File

@@ -22,6 +22,7 @@
#include <cstdlib> #include <cstdlib>
#include <ctime> #include <ctime>
#include <fstream> #include <fstream>
#include <atomic>


#include "common/string_util.h" #include "common/string_util.h"
#include "common/util.h" #include "common/util.h"
@@ -61,6 +62,7 @@ const std::set<domi::FrameworkType> kSupportTensorAsOutput = {
domi::CAFFE, domi::CAFFE,
domi::ONNX domi::ONNX
}; };
std::atomic<uint32_t> graph_name_index {};


static string GetSoPath() { static string GetSoPath() {
Dl_info dl_info; Dl_info dl_info;
@@ -637,7 +639,8 @@ domi::Status AclGrphParseUtil::ParseParamsBeforeGraph(const std::map<AscendStrin


string tmp_name; string tmp_name;
GetAclParams(parser_params, ge::ir_option::OUTPUT, tmp_name); GetAclParams(parser_params, ge::ir_option::OUTPUT, tmp_name);
graph_name = tmp_name.empty() ? (kGraphDefaultName + "_" + ge::parser::CurrentTimeInStr()) : tmp_name;
graph_name = tmp_name.empty() ? (kGraphDefaultName + "_" +
ge::parser::CurrentTimeInStr() + "_" + std::to_string(graph_name_index++)) : tmp_name;


string enable_scope_fusion_passes; string enable_scope_fusion_passes;
GetAclParams(parser_params, ge::ir_option::ENABLE_SCOPE_FUSION_PASSES, enable_scope_fusion_passes); GetAclParams(parser_params, ge::ir_option::ENABLE_SCOPE_FUSION_PASSES, enable_scope_fusion_passes);


Loading…
Cancel
Save