Skip to content

Commit 6e2b600

Browse files
authored
cann: update cmake (#8765)
1 parent c887d8b commit 6e2b600

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

ggml/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ set(GGML_PUBLIC_HEADERS
207207
include/ggml-alloc.h
208208
include/ggml-backend.h
209209
include/ggml-blas.h
210+
include/ggml-cann.h
210211
include/ggml-cuda.h
211212
include/ggml.h
212213
include/ggml-kompute.h

ggml/src/CMakeLists.txt

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -849,11 +849,6 @@ if (GGML_CANN)
849849
${CANN_INSTALL_DIR}/acllib/include
850850
)
851851
852-
# TODO: find libs
853-
link_directories(
854-
${CANN_INSTALL_DIR}/lib64
855-
)
856-
857852
add_subdirectory(ggml-cann/kernels)
858853
list(APPEND CANN_LIBRARIES
859854
ascendcl
@@ -872,6 +867,7 @@ if (GGML_CANN)
872867
873868
set(GGML_EXTRA_LIBS ${GGML_EXTRA_LIBS} ${CANN_LIBRARIES} )
874869
set(GGML_EXTRA_INCLUDES ${GGML_EXTRA_INCLUDES} ${CANN_INCLUDE_DIRS})
870+
set(GGML_EXTRA_LIBDIRS ${GGML_EXTRA_LIBDIRS} ${CANN_INSTALL_DIR}/lib64)
875871
list(APPEND GGML_CDEF_PUBLIC GGML_USE_CANN)
876872
endif()
877873
else()

0 commit comments

Comments
 (0)