Skip to content

Commit 5b351e9

Browse files
authored
cuda : remove nchannels_x argument from mul_mat_vec_nc_f16_f32 (#2028)
- Not used
1 parent 6432aab commit 5b351e9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml-cuda.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1292,7 +1292,7 @@ static __global__ void mul_mat_p021_f16_f32(const void * vx, const float * y, fl
12921292

12931293
static __global__ void mul_mat_vec_nc_f16_f32( // nc == non-contiguous
12941294
const void * vx, const float * y, float * dst, const int ncols_x, const int nrows_x,
1295-
const int row_stride_x, const int nchannels_x, const int channel_stride_x) {
1295+
const int row_stride_x, const int channel_stride_x) {
12961296

12971297
const half * x = (const half *) vx;
12981298

@@ -1698,7 +1698,7 @@ static void ggml_mul_mat_vec_nc_f16_f32_cuda(
16981698
const dim3 block_nums(1, nrows_x, nchannels_x);
16991699
const dim3 block_dims(WARP_SIZE, 1, 1);
17001700
mul_mat_vec_nc_f16_f32<<<block_nums, block_dims, 0, stream>>>
1701-
(vx, y, dst, ncols_x, nrows_x, row_stride_x, nchannels_x, channel_stride_x);
1701+
(vx, y, dst, ncols_x, nrows_x, row_stride_x, channel_stride_x);
17021702
}
17031703

17041704
static void ggml_cpy_f32_f32_cuda(

0 commit comments

Comments
 (0)