Browse Source

GeTensor aligned addr & zero copy support

tags/v1.2.0
chenyemeng 4 years ago
parent
commit
69d0b99385
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      ge/CMakeLists.txt
  2. +1
    -1
      ge/executor/CMakeLists.txt

+ 2
- 2
ge/CMakeLists.txt View File

@@ -125,7 +125,7 @@ set(TRAIN_SRC_LIST
"graph/manager/graph_var_manager.cc"
"graph/manager/host_mem_manager.cc"
"graph/manager/rdma_pool_allocator.cc"
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:graph/manager/host_mem_allocator.cc>
$<$<STREQUAL:${ENABLE_OPEN_SRC},FALSE>:graph/manager/host_mem_allocator.cc>
"graph/manager/memory_api.cc"
"graph/manager/model_manager/event_manager.cc"
"graph/manager/trans_var_data_utils.cc"
@@ -403,7 +403,7 @@ set(INFER_SRC_LIST
"graph/manager/graph_var_manager.cc"
"graph/manager/host_mem_manager.cc"
"graph/manager/rdma_pool_allocator.cc"
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:graph/manager/host_mem_allocator.cc>
$<$<STREQUAL:${ENABLE_OPEN_SRC},FALSE>:graph/manager/host_mem_allocator.cc>
"graph/manager/graph_mem_allocator.cc"
"graph/manager/graph_caching_allocator.cc"
"model/ge_model.cc"


+ 1
- 1
ge/executor/CMakeLists.txt View File

@@ -28,7 +28,7 @@ set(SRC_LIST
"../graph/manager/trans_var_data_utils.cc"
"../graph/manager/util/debug.cc"
"../graph/manager/rdma_pool_allocator.cc"
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:../graph/manager/host_mem_allocator.cc>
$<$<STREQUAL:${ENABLE_OPEN_SRC},FALSE>:../graph/manager/host_mem_allocator.cc>
"../hybrid/node_executor/aicpu/aicpu_ext_info.cc"
"../model/ge_model.cc"
"../model/ge_root_model.cc"


Loading…
Cancel
Save