metal : FA support F32 K and V and head size = 32 (ggml-org#16531)

* metal : FA support F32 K and V and head size = 32

* graph : remove obsolete comment [no ci]

Author: ggerganov

ggml/src/ggml-metal/ggml-metal-device.m

@@ -693,7 +693,8 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
             return true;
         case GGML_OP_FLASH_ATTN_EXT:
             // for new head sizes, add checks here
-            if (op->src[0]->ne[0] != 40 &&
+            if (op->src[0]->ne[0] != 32 &&
+                op->src[0]->ne[0] != 40 &&
                 op->src[0]->ne[0] != 64 &&
                 op->src[0]->ne[0] != 80 &&
                 op->src[0]->ne[0] != 96 &&