!359 Fix tvm copy path

Merge pull request !359 from ZhouFeng/fix-tvm-copy-path
5 years ago · 3511391f35
parent f9dd47620c a66b0c1e11
commit 3511391f35
1 changed files with 3 additions and 3 deletions
--- a/cmake/package.cmake
+++ b/cmake/package.cmake
@ -165,11 +165,11 @@ if (ENABLE_GPU)
        DESTINATION ${INSTALL_PY_DIR}/../
        COMPONENT mindspore
    )
-    if (EXISTS ${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm)
+    if (EXISTS ${CMAKE_BINARY_DIR}/incubator-tvm)
        install(
            DIRECTORY
-                ${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm/topi/python/topi
-                ${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm/python/tvm
+                ${CMAKE_BINARY_DIR}/incubator-tvm/topi/python/topi
+                ${CMAKE_BINARY_DIR}/incubator-tvm/python/tvm
            DESTINATION ${INSTALL_PY_DIR}/../_akg
            COMPONENT mindspore
        )