metal : fix FA support checks (#29122)

This commit is contained in:
Georgi Gerganov
2026-09-19 11:33:03 +03:00
committed by GitHub
parent efa28e950e
commit 60b06ab9a9
+6
View File
@@ -1733,6 +1733,12 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
op->src[0]->ne[0] != 576) {
return false;
}
if (op->src[1]->ne[0] == 72 && op->src[1]->ne[0] != op->src[2]->ne[0]) {
return false;
}
if (op->src[1]->ne[0] < op->src[2]->ne[0]) {
return false;
}
if (op->src[1]->type != op->src[2]->type) {
return false;
}