Function std::cuda_flash_attention_ex

int cuda_flash_attention_ex(
    struct CudaBuf q,
    struct CudaBuf k,
    struct CudaBuf v,
    struct CudaBuf out,
    struct CudaAttnShape s,
    float scale,
    int causal,
)