Skip to content

Commit 652b0f7

Browse files
committed
Merge branch 'npu_support' of github.com:hipudding/llama.cpp into npu_support
2 parents 8342ec4 + a21434a commit 652b0f7

File tree

7 files changed

+330
-22
lines changed

7 files changed

+330
-22
lines changed

CMakeLists.txt

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1218,6 +1218,7 @@ add_library(ggml OBJECT
12181218
ggml-backend.h
12191219
ggml-quants.c
12201220
ggml-quants.h
1221+
<<<<<<< HEAD
12211222
${GGML_SOURCES_CUDA} ${GGML_HEADERS_CUDA}
12221223
${GGML_SOURCES_OPENCL} ${GGML_HEADERS_OPENCL}
12231224
${GGML_SOURCES_METAL} ${GGML_HEADERS_METAL}
@@ -1228,6 +1229,17 @@ add_library(ggml OBJECT
12281229
${GGML_SOURCES_VULKAN} ${GGML_HEADERS_VULKAN}
12291230
${GGML_SOURCES_ROCM} ${GGML_HEADERS_ROCM}
12301231
${GGML_SOURCES_LLAMAFILE} ${GGML_HEADERS_LLAMAFILE}
1232+
=======
1233+
${GGML_SOURCES_CUDA} ${GGML_HEADERS_CUDA}
1234+
${GGML_SOURCES_OPENCL} ${GGML_HEADERS_OPENCL}
1235+
${GGML_SOURCES_METAL} ${GGML_HEADERS_METAL}
1236+
${GGML_SOURCES_MPI} ${GGML_HEADERS_MPI}
1237+
${GGML_SOURCES_EXTRA} ${GGML_HEADERS_EXTRA}
1238+
${GGML_SOURCES_SYCL} ${GGML_HEADERS_SYCL}
1239+
${GGML_SOURCES_KOMPUTE} ${GGML_HEADERS_KOMPUTE}
1240+
${GGML_SOURCES_VULKAN} ${GGML_HEADERS_VULKAN}
1241+
${GGML_SOURCES_ROCM} ${GGML_HEADERS_ROCM}
1242+
>>>>>>> a21434a136a034b6c64bd50b17442e36f3d7e3c8
12311243
${GGML_SOURCES_CANN} ${GGML_HEADERS_CANN}
12321244
)
12331245

examples/llava/clip.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -983,6 +983,11 @@ struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
983983
LOG_TEE("%s: CLIP using Metal backend\n", __func__);
984984
#endif
985985

986+
#ifdef GGML_USE_CANN
987+
new_clip->backend = ggml_backend_cann_init(0);
988+
printf("%s: CLIP using CANN backend\n", __func__);
989+
#endif
990+
986991
#ifdef GGML_USE_CANN
987992
new_clip->backend = ggml_backend_cann_init(0);
988993
printf("%s: CLIP using CANN backend\n", __func__);

ggml-cann.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,6 +450,8 @@ static bool ggml_cann_compute_forward(ggml_backend_cann_context& ctx,
450450
break;
451451
case GGML_OP_ROPE:
452452
case GGML_OP_ALIBI:
453+
ggml_cann_alibi(ctx, dst);
454+
break;
453455
case GGML_OP_IM2COL:
454456
ggml_cann_im2col(ctx, dst);
455457
break;
@@ -685,8 +687,8 @@ GGML_CALL static bool ggml_backend_cann_supports_op(ggml_backend_t backend,
685687
return true;
686688
case GGML_OP_SOFT_MAX:
687689
case GGML_OP_ROPE:
688-
case GGML_OP_ALIBI:
689690
return false;
691+
case GGML_OP_ALIBI:
690692
case GGML_OP_IM2COL:
691693
return true;
692694
case GGML_OP_POOL_2D:

0 commit comments

Comments
 (0)