pub fn build_mha_fwd_graph( dtype: DtypeTag, p: &AttentionParams, layout: TensorLayout, ) -> OperationGraphSpec