vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test-backend-ops : check all the ops in the test for support in the backends

Browse source
This commit is contained in:
slaren 2024-01-26 20:01:36 +01:00
parent 2512799cfe
commit 8ca33dec7d
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test-backend-ops.cpp
View file

@ -364,15 +364,15 @@ struct test_case {
printf(" %s(%s): ", op_desc(out).c_str(), vars().c_str()); printf(" %s(%s): ", op_desc(out).c_str(), vars().c_str());
fflush(stdout); fflush(stdout);
// check if backends support op // check if the backends support the ops
bool supported = true; bool supported = true;
for (ggml_backend_t backend : {backend1, backend2}) { for (ggml_backend_t backend : {backend1, backend2}) {
if ( for (ggml_tensor * t = ggml_get_first_tensor(ctx); t != NULL; t = ggml_get_next_tensor(ctx, t)) {
!ggml_backend_supports_op(backend, out) if (!ggml_backend_supports_op(backend, t)) {
|| (op_desc(out) == "MOE" && !strcmp(ggml_backend_name(backend), "Kompute")) printf("not supported [%s] ", ggml_backend_name(backend));
) { supported = false;
printf("not supported [%s] ", ggml_backend_name(backend)); break;
supported = false; }
} }
} }
if (!supported) { if (!supported) {