diff --git a/cmake/package.cmake b/cmake/package.cmake index 7c9c92b23c96be08fa57c60aa5b9bfa53d83bbd7..d04efca085cbbb2b3ecde2ade5a4125ff24260ba 100644 --- a/cmake/package.cmake +++ b/cmake/package.cmake @@ -90,19 +90,23 @@ if (ENABLE_CPU) endif () if (ENABLE_GPU) - if (ENABLE_MPI) - install( - TARGETS _ms_mpi gpu_collective - DESTINATION ${INSTALL_LIB_DIR} - COMPONENT mindspore - ) - endif () - - install( - TARGETS gpu_queue - DESTINATION ${INSTALL_LIB_DIR} - COMPONENT mindspore - ) + if (ENABLE_MPI) + install( + TARGETS _ms_mpi + DESTINATION ${INSTALL_BASE_DIR} + COMPONENT mindspore + ) + install( + TARGETS gpu_collective + DESTINATION ${INSTALL_LIB_DIR} + COMPONENT mindspore + ) + endif () + install( + TARGETS gpu_queue + DESTINATION ${INSTALL_LIB_DIR} + COMPONENT mindspore + ) endif () if (NOT ENABLE_GE)