Function std::mps_flash_attention_ex

int mps_flash_attention_ex(
    struct MpsBuf q,
    struct MpsBuf k,
    struct MpsBuf v,
    struct MpsBuf out,
    struct MpsAttnShape s,
    int dtype,
    float scale,
    int causal,
)