metal : add FA specialization for HSK = 320, HSV = 256 (#20549)

2026-03-14 23:15:47 +02:00
parent b4768955c4
commit b30a5fdf37
3 changed files with 26 additions and 4 deletions
@@ -1142,6 +1142,7 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
                op->src[0]->ne[0] != 128 &&
                op->src[0]->ne[0] != 192 &&
                op->src[0]->ne[0] != 256 &&
+                op->src[0]->ne[0] != 320 &&
                op->src[0]->ne[0] != 576) {
                return false;
            }