happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

CANN: Disable Ger operator of OUT_PROD on 310p device (#17563)

This commit is contained in:
TianHao324 2025-12-02 20:35:23 +08:00 committed by GitHub
parent 7f3a72a8ed
commit a2b0fe8d37
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ggml/src/ggml-cann/ggml-cann.cpp
View File

@ -2564,6 +2564,10 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
return true; return true;
case GGML_OP_OUT_PROD: case GGML_OP_OUT_PROD:
{ {
#ifdef ASCEND_310P
// Ger is not supported on 310p device
return false;
#endif
switch (op->src[0]->type) { switch (op->src[0]->type) {
case GGML_TYPE_F16: case GGML_TYPE_F16:
case GGML_TYPE_F32: case GGML_TYPE_F32: