Skip to content

Commit f15167a

Browse files
committed
cuda : do not use dmmv if the tensor does not have enough cols
1 parent 713665d commit f15167a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ggml/src/ggml-cuda.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1875,7 +1875,8 @@ static void ggml_cuda_mul_mat(ggml_backend_cuda_context & ctx, const ggml_tensor
18751875

18761876
bool use_dequantize_mul_mat_vec = (ggml_is_quantized(src0->type) || src0->type == GGML_TYPE_F16)
18771877
&& src1->type == GGML_TYPE_F32 && dst->type == GGML_TYPE_F32
1878-
&& src0->ne[0] % GGML_CUDA_DMMV_X == 0 && src1->ne[1] == 1;
1878+
&& src0->ne[0] % GGML_CUDA_DMMV_X == 0 && src0->ne[0] >= GGML_CUDA_DMMV_X*2
1879+
&& src1->ne[1] == 1;
18791880
bool use_mul_mat_vec_q = ggml_is_quantized(src0->type)
18801881
&& src1->type == GGML_TYPE_F32 && dst->type == GGML_TYPE_F32
18811882
&& src1->ne[1] <= MMVQ_MAX_BATCH_SIZE;

0 commit comments

Comments
 (0)