Flash Attention Triton

• easydel.kernels.gpu_ops.flash_attention_triton.__init__
  • triton_flash_attention()
• easydel.kernels.gpu_ops.flash_attention_triton._backward_triton
  • config_prune_kernel()
• easydel.kernels.gpu_ops.flash_attention_triton._flash_attention
  • flash_attention()
• easydel.kernels.gpu_ops.flash_attention_triton._forward_triton
  • config_prune_kernel()
• easydel.kernels.gpu_ops.flash_attention_triton._utils
  • attention_pack_with_static_shape()
  • attention_unpack_with_static_shape()
  • basic_attention_refrence()
  • calc_bias_strides()