metal : add set_rows with src0 f16 (#25434)

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
fairydreaming
2026-07-08 09:49:07 +03:00
committed by GitHub
co-authored by Georgi Gerganov
parent 4a7ee3126d
commit 0512ef1e5a
5 changed files with 83 additions and 80 deletions
+1 -1
View File
@@ -1334,7 +1334,7 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
return op->src[0]->type != GGML_TYPE_NVFP4;
case GGML_OP_SET_ROWS:
{
if (op->src[0]->type != GGML_TYPE_F32) {
if (op->src[0]->type != GGML_TYPE_F32 && op->src[0]->type != GGML_TYPE_F16) {
return false;
}