Browse Source

GeTensor aligned addr & zero copy support

tags/v1.2.0
chenyemeng 4 years ago
parent
commit
5486fff87a
2 changed files with 4 additions and 1 deletions
  1. +1
    -1
      ge/graph/manager/host_mem_manager.h
  2. +3
    -0
      ge/graph/preprocess/graph_preprocess.cc

+ 1
- 1
ge/graph/manager/host_mem_manager.h View File

@@ -42,7 +42,7 @@ struct SharedMemInfo {
uint64_t mem_size = 0;
int fd = 0;
uint8_t *device_address = nullptr;
#if !ENABLE_OPEN_SRC
#if (ENABLE_OPEN_SRC != True)
std::shared_ptr<AlignedPtr> host_aligned_ptr = nullptr;
#else
uint8_t *host_address = nullptr;


+ 3
- 0
ge/graph/preprocess/graph_preprocess.cc View File

@@ -37,6 +37,9 @@
#include "graph/passes/addn_pass.h"
#include "graph/passes/aicpu_constant_folding_pass.h"
#include "graph/passes/assert_pass.h"
#if (ENABLE_OPEN_SRC == True)
#include "graph/passes/assign_pass.h"
#endif
#include "graph/passes/common_subexpression_elimination_pass.h"
#include "graph/passes/cond_pass.h"
#include "graph/passes/cond_remove_pass.h"


Loading…
Cancel
Save