Skip to content

Commit fdad5eb

Browse files
committed
Update set-rows.cu
1 parent 42a1ae6 commit fdad5eb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

ggml/src/ggml-cuda/set-rows.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ static void set_rows_cuda(ggml_backend_cuda_context & ctx, const ggml_tensor * s
236236
stream
237237
);
238238
} else if (dst->type == GGML_TYPE_Q6_0) {
239-
set_rows_cuda_quant<block_q6_0, QK6_0, quantize_f32_q6_0_block>(
239+
set_rows_cuda_quant<idx_t, block_q6_0, QK6_0, quantize_f32_q6_0_block>(
240240
src0_d, src1_d, (block_q6_0*)dst->data,
241241
ne00, ne01, ne02, ne03,
242242
ne10, ne11, ne12, ne13,

0 commit comments

Comments
 (0)