|
|
|
@@ -162,7 +162,7 @@ install( |
|
|
|
if (ENABLE_GPU) |
|
|
|
install( |
|
|
|
DIRECTORY ${CMAKE_SOURCE_DIR}/mindspore/_akg |
|
|
|
DESTINATION ${INSTALL_PY_DIR} |
|
|
|
DESTINATION ${INSTALL_PY_DIR}/../ |
|
|
|
COMPONENT mindspore |
|
|
|
) |
|
|
|
if (EXISTS ${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm) |
|
|
|
@@ -170,7 +170,7 @@ if (ENABLE_GPU) |
|
|
|
DIRECTORY |
|
|
|
${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm/topi/python/topi |
|
|
|
${CMAKE_SOURCE_DIR}/mindspore/incubator-tvm/python/tvm |
|
|
|
DESTINATION ${INSTALL_PY_DIR}/_akg |
|
|
|
DESTINATION ${INSTALL_PY_DIR}/../_akg |
|
|
|
COMPONENT mindspore |
|
|
|
) |
|
|
|
endif () |
|
|
|
|