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:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user