From 9d1887240b27a4ff76984b18600844ebca8604be Mon Sep 17 00:00:00 2001 From: wjm Date: Wed, 10 Feb 2021 02:32:28 +0800 Subject: [PATCH] fix train --- CMakeLists.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index d887c87..8cf57ee 100644 --- a/CMakeLists.txt +++ b/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()