metal: add support for opt_step_sgd (#16539)

* metal: add support for opt_step_sgd

* add newline to pass EditorConfig check
This commit is contained in:
Sam/Samuel
2025-10-13 16:25:02 +08:00
committed by GitHub
parent c515fc5771
commit 3f750f8d76
7 changed files with 78 additions and 0 deletions

View File

@@ -800,6 +800,7 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
};
}
case GGML_OP_OPT_STEP_ADAMW:
case GGML_OP_OPT_STEP_SGD:
return has_simdgroup_reduction;
default:
return false;