Browse Source

!1968 Sort CMakeLists.txt SRC_LIST

Merge pull request !1968 from 张晓昆/master
tags/v1.5.1
i-robot Gitee 3 years ago
parent
commit
b84e1438b6
3 changed files with 624 additions and 512 deletions
  1. +623
    -479
      ge/CMakeLists.txt
  2. +0
    -1
      ge/graph/manager/graph_manager.h
  3. +1
    -32
      tests/ut/ge/graph/load/model_manager_unittest.cc

+ 623
- 479
ge/CMakeLists.txt
File diff suppressed because it is too large
View File


+ 0
- 1
ge/graph/manager/graph_manager.h View File

@@ -32,7 +32,6 @@
#include "external/ge/ge_api_types.h" #include "external/ge/ge_api_types.h"
#include "graph/build/graph_builder.h" #include "graph/build/graph_builder.h"
#include "graph/ge_local_context.h" #include "graph/ge_local_context.h"
#include "graph/load/graph_loader.h"
#include "graph/manager/graph_manager_utils.h" #include "graph/manager/graph_manager_utils.h"
#include "graph/manager/util/variable_accelerate_ctrl.h" #include "graph/manager/util/variable_accelerate_ctrl.h"
#include "graph/optimize/graph_optimize.h" #include "graph/optimize/graph_optimize.h"


+ 1
- 32
tests/ut/ge/graph/load/model_manager_unittest.cc View File

@@ -54,31 +54,13 @@ class UtestModelManagerModelManager : public testing::Test {
} }


void SetUp() {} void SetUp() {}

void TearDown() {} void TearDown() {}


void CreateGraph(Graph &graph) {
TensorDesc desc(ge::Shape({1, 3, 224, 224}));
uint32_t size = desc.GetShape().GetShapeSize();
desc.SetSize(size);
auto data = op::Data("Data").set_attr_index(0);
data.update_input_desc_data(desc);
data.update_output_desc_out(desc);

auto flatten = op::Flatten("Flatten").set_input_x(data, data.name_out_out());

std::vector<Operator> inputs{data};
std::vector<Operator> outputs{flatten};
std::vector<Operator> targets{flatten};
// Graph graph("test_graph");
graph.SetInputs(inputs).SetOutputs(outputs).SetTargets(targets);
}

void GenUnencryptModelData(ModelData &data) { void GenUnencryptModelData(ModelData &data) {
const int model_len = 10; const int model_len = 10;
data.model_len = sizeof(ModelFileHeader) + model_len; data.model_len = sizeof(ModelFileHeader) + model_len;
data.model_data = new uint8_t[data.model_len]; data.model_data = new uint8_t[data.model_len];
memset((uint8_t *)data.model_data + sizeof(ModelFileHeader), 0, model_len);
memset(data.model_data, 0, data.model_len);


ModelFileHeader *header = (ModelFileHeader *)data.model_data; ModelFileHeader *header = (ModelFileHeader *)data.model_data;
header->magic = MODEL_FILE_MAGIC_NUM; header->magic = MODEL_FILE_MAGIC_NUM;
@@ -88,19 +70,6 @@ class UtestModelManagerModelManager : public testing::Test {
header->is_checksum = ModelCheckType::CHECK; header->is_checksum = ModelCheckType::CHECK;
} }


void GenEncryptModelData(ModelData &data) {
const int model_len = 10;
data.key = ENC_KEY;
data.model_data = new uint8_t[data.model_len];
uint8_t data_ori[model_len];
memset(data_ori, 0, model_len);
ModelFileHeader *header = (ModelFileHeader *)data.model_data;
header->magic = MODEL_FILE_MAGIC_NUM;
header->version = MODEL_VERSION;
header->is_encrypt = ModelEncryptType::ENCRYPTED;
header->length = 10; // encrypt_len;
}

void LoadStandardModelData(ModelData &data) { void LoadStandardModelData(ModelData &data) {
data.model_len = 512; data.model_len = 512;
data.model_data = new uint8_t[data.model_len]; data.model_data = new uint8_t[data.model_len];


Loading…
Cancel
Save