fix train

CMakeLists.txt

@@ -50,12 +50,13 @@ if (ENABLE_OPEN_SRC)
         endif()
 
         set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64)
+        set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64)
 
         find_module(slog libalog.so ${ASCEND_ATC_DIR})
         find_module(static_mmpa libmmpa.a ${ASCEND_ATC_DIR})
         find_module(error_manager liberror_manager.so ${ASCEND_ATC_DIR})
         find_module(ge_compiler libge_compiler.so ${ASCEND_ATC_DIR})
-        find_module(ge_runner libge_runner.so ${ASCEND_ATC_DIR})
+        find_module(ge_runner libge_runner.so ${ASCEND_RUNTIME_DIR})
         find_module(ge_common libge_common.so ${ASCEND_ATC_DIR})
     endif()