--- a/ggml/src/ggml-cuda/fattn.cu +++ b/ggml/src/ggml-cuda/fattn.cu @@ -175,7 +175,8 @@ } if constexpr (ncols2 <= 16) { - if (Q->ne[1] <= 16/ncols2) { + // Volta MMA kernels require at least 32 columns. + if (!volta_mma_available(cc) && Q->ne[1] <= 16/ncols2) { ggml_cuda_flash_attn_ext_mma_f16_case(ctx, dst); return; }