Gpu Ops

• Deprecated
  • easydel.kernels.gpu_ops._deprecated.pallas_gemm
  • easydel.kernels.gpu_ops._deprecated.pallas_gqa_flash_attention_2
  • easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2
  • easydel.kernels.gpu_ops._deprecated.triton_gqa_flash_attention_2
• easydel.kernels.gpu_ops.__init__
  • triton_flash_attention()
• easydel.kernels.gpu_ops._utils
  • dtype_index()
  • get_sharding()
  • get_strides()
  • safe_autotune()
• Flash Attention Triton
  • easydel.kernels.gpu_ops.flash_attention_triton.__init__
  • easydel.kernels.gpu_ops.flash_attention_triton._backward_triton
  • easydel.kernels.gpu_ops.flash_attention_triton._flash_attention
  • easydel.kernels.gpu_ops.flash_attention_triton._forward_triton
  • easydel.kernels.gpu_ops.flash_attention_triton._utils
• Matmul Triton
  • easydel.kernels.gpu_ops.matmul_triton.__init__
  • easydel.kernels.gpu_ops.matmul_triton._caller