flash-attention/csrc/composable_kernel/include/ck/README.md

Back to the main page

Composable Kernel supported operations

Supported device operations

• Average pooling
• Batched contraction
• Batched gemm
• Batchnorm
• CGEMM
• Contraction
• Convolution
• Image to Column and Column to Image
• Elementwise
• GEMM
• Max pooling
• Reduce
• Normalization
• Permute
• Put
• Softmax