提交 109a21d5 编写于 作者: T tronzhang

Adapt change of module akg

上级 e5193176
...@@ -84,20 +84,7 @@ if (ENABLE_GE OR ENABLE_D OR ENABLE_TESTCASES) ...@@ -84,20 +84,7 @@ if (ENABLE_GE OR ENABLE_D OR ENABLE_TESTCASES)
endif() endif()
if (ENABLE_AKG AND ENABLE_D) if (ENABLE_AKG AND ENABLE_D)
set(AKG_PATH "${CMAKE_SOURCE_DIR}/akg/mindspore/ccsrc/akg") add_subdirectory("${CMAKE_SOURCE_DIR}/akg")
add_subdirectory(${AKG_PATH})
set(TVM_PATH "${CMAKE_CURRENT_BINARY_DIR}/akg/mindspore/ccsrc/akg/incubator-tvm")
include_directories("${TVM_PATH}/include")
include_directories("${TVM_PATH}")
include_directories("${TVM_PATH}/src")
include_directories("${TVM_PATH}/src/schedule")
include_directories("${TVM_PATH}/3rdparty/dmlc-core/include")
include_directories("${TVM_PATH}/3rdparty/dlpack/include")
include_directories("${TVM_PATH}/3rdparty/compiler-rt")
include_directories("${TVM_PATH}/3rdparty/rang/include")
include_directories("${TVM_PATH}/3rdparty/picojson")
include_directories("${AKG_PATH}")
include_directories("${AKG_PATH}/include")
endif() endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden")
......
akg @ f257d67b
Subproject commit 02133cc76374b8c2d6c73f24a58d3c28e1e4d14e Subproject commit f257d67b1e869b3a613cb9806b73ddc36cfe86c8
...@@ -454,11 +454,4 @@ fi ...@@ -454,11 +454,4 @@ fi
cp -rf ${BUILD_PATH}/package/mindspore/lib ${BUILD_PATH}/../mindspore cp -rf ${BUILD_PATH}/package/mindspore/lib ${BUILD_PATH}/../mindspore
cp -rf ${BUILD_PATH}/package/mindspore/*.so ${BUILD_PATH}/../mindspore cp -rf ${BUILD_PATH}/package/mindspore/*.so ${BUILD_PATH}/../mindspore
if [[ "X$ENABLE_AKG" = "Xon" ]] && [[ "X$ENABLE_D" = "Xon" ]]; then
so_lib_dir=${BUILD_PATH}/package/mindspore/lib
akg_build_dir=${BUILD_PATH}/mindspore/akg/mindspore/ccsrc/akg
mkdir -p ${so_lib_dir}
cp ${akg_build_dir}/*.so ${so_lib_dir}
fi
echo "---------------- mindspore: build end ----------------" echo "---------------- mindspore: build end ----------------"
...@@ -223,21 +223,13 @@ if (ENABLE_GPU) ...@@ -223,21 +223,13 @@ if (ENABLE_GPU)
endif () endif ()
if (ENABLE_D AND ENABLE_AKG) if (ENABLE_D AND ENABLE_AKG)
set (AKG_PATH ${CMAKE_SOURCE_DIR}/akg) set (AKG_PATH ${CMAKE_SOURCE_DIR}/build/mindspore/akg)
set (TVM_PATH ${CMAKE_SOURCE_DIR}/build/mindspore/akg/mindspore/ccsrc/akg/incubator-tvm)
install( install(
DIRECTORY DIRECTORY
${AKG_PATH}/mindspore/akg ${AKG_PATH}/akg
DESTINATION ${INSTALL_PY_DIR}/.. DESTINATION ${INSTALL_PY_DIR}/..
COMPONENT mindspore COMPONENT mindspore
) )
install(
DIRECTORY
${TVM_PATH}/topi/python/topi
${TVM_PATH}/python/tvm
DESTINATION ${INSTALL_PY_DIR}/../akg
COMPONENT mindspore
)
endif() endif()
if (EXISTS ${CMAKE_SOURCE_DIR}/mindspore/dataset) if (EXISTS ${CMAKE_SOURCE_DIR}/mindspore/dataset)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册