easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2

easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2.pallas_mha_flash_attention2_gpu(q: Array, k: Array, v: Array, b: Optional[Array] = None, dtype: dtype = None, qblock: int = 128, kblock: int = 128, softmax_scale: float = None)