From cb3cc7993e984d23e3f94756a0e0b94106a88b1b Mon Sep 17 00:00:00 2001
From: taoxiangdong <taoxiangdong1@huawei.com>
Date: Fri, 23 Oct 2020 18:47:37 +0800
Subject: [PATCH] update cmakelist

---
 CMakeLists.txt | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index f8d419bc..d37cbd45 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -62,12 +62,13 @@ if (ENABLE_OPEN_SRC)
         endif()
         set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64)
         set(ASCEND_DRIVER_COMMON_DIR ${ASCEND_DIR}/driver/lib64/common)
+        set(ASCEND_DRIVER_SHARE_DIR ${ASCEND_DIR}/driver/lib64/share)
         set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64)
         set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64)
         set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64)
+        find_module(slog libslog.so ${ASCEND_ATC_DIR})
+        find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
         if(PLATFORM STREQUAL "train")
-            find_module(slog libslog.so ${ASCEND_ATC_DIR})
-            find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
             find_module(msprof libmsprof.so ${ASCEND_DRIVER_COMMON_DIR})
             find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})
             find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR})
@@ -80,8 +81,6 @@ if (ENABLE_OPEN_SRC)
                 message(FATAL_ERROR "This platform is not supported in train mode, build terminated")
             endif()
         elseif(PLATFORM STREQUAL "inference")
-            find_module(slog libslog.so ${ASCEND_ATC_DIR})
-            find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
             find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR})
             find_module(runtime libruntime.so ${ASCEND_ACL_DIR})
             find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
@@ -94,21 +93,19 @@ if (ENABLE_OPEN_SRC)
                 find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
                 find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})
             else()
-                find_module(msprof libmsprof.so ${ASCEND_ATC_DIR})
+                find_module(msprof libmsprof.so ${ASCEND_DRIVER_SHARE_DIR})
             endif()
         elseif(PLATFORM STREQUAL "all")
-            find_module(slog libslog.so ${ASCEND_DRIVER_DIR})
-            find_module(mmpa libmmpa.so ${ASCEND_DRIVER_DIR})
-            find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})
-            find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})
-            find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR})
-            find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR})
+            find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})#mdc没有 dc在driver目录
+            find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})#mdc没有 dc在fwkack目录
+            find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR})
+            find_module(runtime libruntime.so ${ASCEND_ACL_DIR})
             find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
-            find_module(resource libresource.so ${ASCEND_RUNTIME_DIR})
-            find_module(error_manager liberror_manager.so ${ASCEND_RUNTIME_DIR})
+            find_module(resource libresource.so ${ASCEND_ATC_DIR})
+            find_module(error_manager liberror_manager.so ${ASCEND_ATC_DIR})
             find_module(error_manager_static liberror_manager.a ${ASCEND_ACL_DIR})
-            find_module(msprofiler libmsprofiler.a ${ASCEND_RUNTIME_DIR})
-            find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
+            find_module(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR})
+            find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver)#mdc没有 dc在driver/lib64/driver
             find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR})
         else()
              message(FATAL_ERROR "PLATFORM param is invalid, should be train or inference, build terminated")