!368 Fix _ms_mpi package path

Merge pull request !368 from ZhouFeng/fix-pkg-path
5 years ago · 7d406e8e6c
parent 58b013c319 108ac9cda7
commit 7d406e8e6c
1 changed files with 17 additions and 13 deletions
--- 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)