We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c0389db commit 2b3efeaCopy full SHA for 2b3efea
ggml/src/ggml-cpu/kleidiai/kleidiai.cpp
@@ -515,9 +515,6 @@ class extra_buffer_type : ggml::cpu::extra_buffer_type {
515
op->src[0]->buffer &&
516
(ggml_n_dims(op->src[0]) == 2) &&
517
op->src[0]->buffer->buft == ggml_backend_cpu_kleidiai_buffer_type() && ctx.kernels) {
518
- if (op->op == GGML_OP_GET_ROWS && op->src[1]->ne[0] != 8) {
519
- return false;
520
- }
521
if (op->src[1]->buffer && !ggml_backend_buft_is_host(op->src[1]->buffer->buft)) {
522
return false;
523
}
0 commit comments