Browse Source

!20936 add version of mindir

Merge pull request !20936 from lianliguang/add_version_of_mind_ir
tags/v1.4.0
i-robot Gitee 4 years ago
parent
commit
7b8a82acf4
2 changed files with 7 additions and 3 deletions
  1. +2
    -0
      mindspore/ccsrc/transform/express_ir/CMakeLists.txt
  2. +5
    -3
      mindspore/ccsrc/transform/express_ir/mindir_exporter.cc

+ 2
- 0
mindspore/ccsrc/transform/express_ir/CMakeLists.txt View File

@@ -1,4 +1,6 @@
file(GLOB_RECURSE _EXPORTER_IR_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
file(STRINGS "${CMAKE_SOURCE_DIR}/version.txt" VERSION)
add_definitions(-DVERSION=\"${VERSION}\")
set_property(SOURCE ${_EXPORTER_IR_SRC_FILES} PROPERTY COMPILE_DEFINITIONS
SUBMODULE_ID=mindspore::SubModuleId::SM_EXPRESS)
add_library(_mindspore_transform_express_ir_obj OBJECT ${_EXPORTER_IR_SRC_FILES})

+ 5
- 3
mindspore/ccsrc/transform/express_ir/mindir_exporter.cc View File

@@ -170,9 +170,11 @@ std::string IrExportBuilder::GetProtoString(const FuncGraphPtr &func_graph) {
}

void IrExportBuilder::BuildModelInfo() {
model_.set_ir_version("0.1.0");
model_.set_producer_name("MindSpore");
model_.set_model_version("1.1.0");
constexpr auto ir_version = "0.1.0";
constexpr auto mindspore_name = "MindSpore";
model_.set_ir_version(ir_version);
model_.set_producer_name(mindspore_name);
model_.set_model_version(VERSION);
}

void IrExportBuilder::BuildModel(const FuncGraphPtr &func_graph, bool save_tensor_data) {


Loading…
Cancel
Save