Browse Source

Sort CMakeLists.txt SRC_LIST

tags/v1.5.1
zhangxiaokun 3 years ago
parent
commit
d642976cb6
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 "graph/build/graph_builder.h"
#include "graph/ge_local_context.h"
#include "graph/load/graph_loader.h"
#include "graph/manager/graph_manager_utils.h"
#include "graph/manager/util/variable_accelerate_ctrl.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 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) {
const int model_len = 10;
data.model_len = sizeof(ModelFileHeader) + 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;
header->magic = MODEL_FILE_MAGIC_NUM;
@@ -88,19 +70,6 @@ class UtestModelManagerModelManager : public testing::Test {
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) {
data.model_len = 512;
data.model_data = new uint8_t[data.model_len];


Loading…
Cancel
Save