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; return FAILED;
} }
AddRePassNode(node); AddRePassNode(node);
break;
} }


GELOGD("InplaceSupportCheckPass success"); 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 target_compile_definitions(ge_load_common PRIVATE
google=ascend_private google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
) )


target_link_libraries(ge_load_common PRIVATE 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 target_compile_definitions(ge_execute_common PRIVATE
google=ascend_private google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
) )


target_link_libraries(ge_execute_common PRIVATE 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 target_compile_definitions(ut_libge_multiparts_utest PRIVATE
google=ascend_private google=ascend_private
$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
) )


target_link_libraries(ut_libge_multiparts_utest target_link_libraries(ut_libge_multiparts_utest


Loading…
Cancel
Save