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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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;
}