File tree Expand file tree Collapse file tree 2 files changed +2
-5
lines changed Expand file tree Collapse file tree 2 files changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -207,6 +207,7 @@ set(GGML_PUBLIC_HEADERS
207
207
include /ggml-alloc.h
208
208
include /ggml-backend.h
209
209
include /ggml-blas.h
210
+ include /ggml-cann.h
210
211
include /ggml-cuda.h
211
212
include /ggml.h
212
213
include /ggml-kompute.h
Original file line number Diff line number Diff line change @@ -849,11 +849,6 @@ if (GGML_CANN)
849
849
${CANN_INSTALL_DIR} /acllib/include
850
850
)
851
851
852
- # TODO: find libs
853
- link_directories(
854
- ${CANN_INSTALL_DIR} /lib64
855
- )
856
-
857
852
add_subdirectory(ggml-cann/kernels)
858
853
list(APPEND CANN_LIBRARIES
859
854
ascendcl
@@ -872,6 +867,7 @@ if (GGML_CANN)
872
867
873
868
set(GGML_EXTRA_LIBS ${GGML_EXTRA_LIBS} ${CANN_LIBRARIES} )
874
869
set(GGML_EXTRA_INCLUDES ${GGML_EXTRA_INCLUDES} ${CANN_INCLUDE_DIRS} )
870
+ set(GGML_EXTRA_LIBDIRS ${GGML_EXTRA_LIBDIRS} ${CANN_INSTALL_DIR} /lib64)
875
871
list(APPEND GGML_CDEF_PUBLIC GGML_USE_CANN)
876
872
endif()
877
873
else()
You can’t perform that action at this time.
0 commit comments