提交 2d35511d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2423 Adapt module akg's change

Merge pull request !2423 from TronZhang/r0.3
......@@ -84,20 +84,7 @@ if (ENABLE_GE OR ENABLE_D OR ENABLE_TESTCASES)
endif()
if (ENABLE_AKG AND ENABLE_D)
set(AKG_PATH "${CMAKE_SOURCE_DIR}/akg/mindspore/ccsrc/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")
add_subdirectory("${CMAKE_SOURCE_DIR}/akg")
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden")
......
akg @ f257d67b
Subproject commit 02133cc76374b8c2d6c73f24a58d3c28e1e4d14e
Subproject commit f257d67b1e869b3a613cb9806b73ddc36cfe86c8
......@@ -454,11 +454,4 @@ fi
cp -rf ${BUILD_PATH}/package/mindspore/lib ${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 ----------------"
......@@ -223,21 +223,13 @@ if (ENABLE_GPU)
endif ()
if (ENABLE_D AND ENABLE_AKG)
set (AKG_PATH ${CMAKE_SOURCE_DIR}/akg)
set (TVM_PATH ${CMAKE_SOURCE_DIR}/build/mindspore/akg/mindspore/ccsrc/akg/incubator-tvm)
set (AKG_PATH ${CMAKE_SOURCE_DIR}/build/mindspore/akg)
install(
DIRECTORY
${AKG_PATH}/mindspore/akg
${AKG_PATH}/akg
DESTINATION ${INSTALL_PY_DIR}/..
COMPONENT mindspore
)
install(
DIRECTORY
${TVM_PATH}/topi/python/topi
${TVM_PATH}/python/tvm
DESTINATION ${INSTALL_PY_DIR}/../akg
COMPONENT mindspore
)
endif()
if (EXISTS ${CMAKE_SOURCE_DIR}/mindspore/dataset)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册