fix: fix assertion

This commit is contained in:
hongruichen 2024-11-29 23:37:52 +08:00
parent 09efaa389e
commit c5e6549331
2 changed files with 3 additions and 1 deletions

View File

@ -666,6 +666,8 @@ bool ggml_qnn_supports_op(ggml_backend_qnn_device_context *ctx, const ggml_tenso
case GGML_OP_MUL_MAT:
return ggml_qnn_supports_matmul_op(ctx, op);
case GGML_OP_VIEW:
return true;
default:
return false;
}

View File

@ -227,7 +227,7 @@ size_t ggml_backend_qnn_buffer_type_get_alignment(ggml_backend_buffer_type_t buf
size_t ggml_backend_qnn_buffer_type_get_max_size(ggml_backend_buffer_type_t buft) {
GGML_UNUSED(buft);
// TODO: get the max size from device
return (1024 * 1024 * 1024);
return (2 * 1024 * 1024 * 1024);
}
bool ggml_backend_qnn_buffer_is_host(ggml_backend_buffer_type_t buft) {