fix: fix assertion

chraac · Nov 29, 2024 · c5e6549 · c5e6549
1 parent 09efaa3
commit c5e6549
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ggml/src/ggml-qnn/backend-ops.cpp b/ggml/src/ggml-qnn/backend-ops.cpp
@@ -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;
         }

diff --git a/ggml/src/ggml-qnn/ggml-qnn.cpp b/ggml/src/ggml-qnn/ggml-qnn.cpp
@@ -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) {