Browse Source

!2423 Adapt module akg's change

Merge pull request !2423 from TronZhang/r0.3
tags/v0.3.1-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
2d35511d7c
4 changed files with 4 additions and 32 deletions
  1. +1
    -14
      CMakeLists.txt
  2. +1
    -1
      akg
  3. +0
    -7
      build.sh
  4. +2
    -10
      cmake/package.cmake

+ 1
- 14
CMakeLists.txt View File

@@ -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")


+ 1
- 1
akg

@@ -1 +1 @@
Subproject commit 02133cc76374b8c2d6c73f24a58d3c28e1e4d14e
Subproject commit f257d67b1e869b3a613cb9806b73ddc36cfe86c8

+ 0
- 7
build.sh View File

@@ -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 ----------------"

+ 2
- 10
cmake/package.cmake View File

@@ -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)


Loading…
Cancel
Save