Browse Source

GeTensor aligned addr & zero copy support

tags/v1.2.0
chenyemeng 3 years ago
parent
commit
84c6c10ceb
2 changed files with 3 additions and 1 deletions
  1. +1
    -0
      ge/graph/passes/inplace_support_check_pass.cc
  2. +2
    -1
      tests/ut/ge/CMakeLists.txt

+ 1
- 0
ge/graph/passes/inplace_support_check_pass.cc View File

@@ -78,6 +78,7 @@ Status InplaceSupportCheckPass::Run(NodePtr &node) {
return FAILED;
}
AddRePassNode(node);
break;
}

GELOGD("InplaceSupportCheckPass success");


+ 2
- 1
tests/ut/ge/CMakeLists.txt View File

@@ -776,6 +776,7 @@ add_library(ge_load_common STATIC ${GRAPH_LOAD_COMMON_SRC_FILES} ${PROTO_SRCS} $

target_compile_definitions(ge_load_common PRIVATE
google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
)

target_link_libraries(ge_load_common PRIVATE
@@ -790,6 +791,7 @@ add_library(ge_execute_common STATIC ${GRAPH_EXECUTE_COMMON_SRC_FILES} ${PROTO_S

target_compile_definitions(ge_execute_common PRIVATE
google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
)

target_link_libraries(ge_execute_common PRIVATE
@@ -856,7 +858,6 @@ target_compile_options(ut_libge_multiparts_utest PRIVATE

target_compile_definitions(ut_libge_multiparts_utest PRIVATE
google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
)

target_link_libraries(ut_libge_multiparts_utest


Loading…
Cancel
Save