GeTensor aligned addr & zero copy support

4 years ago · d4629babd8
parent 9679184b8f
commit d4629babd8
1 changed files with 1 additions and 0 deletions
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@ -856,6 +856,7 @@ 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