fix supported ops for kompute backend

This commit is contained in:
Jared Van Bortel 2024-01-18 15:32:55 -05:00
parent 7addf2b878
commit 610394fff8
2 changed files with 30 additions and 16 deletions

View file

@ -360,7 +360,10 @@ struct test_case {
// check if backends support op
bool supported = true;
for (ggml_backend_t backend : {backend1, backend2}) {
if (!ggml_backend_supports_op(backend, out)) {
if (
!ggml_backend_supports_op(backend, out)
|| (op_desc(out) == "MOE" && !strcmp(ggml_backend_name(backend), "Kompute"))
) {
printf("not supported [%s] ", ggml_backend_name(backend));
supported = false;
}