Skip to content

Commit

Permalink
cuda : remove nchannels_x argument from mul_mat_vec_nc_f16_f32 (ggerg…
Browse files Browse the repository at this point in the history
…anov#2028)

- Not used
  • Loading branch information
set-soft authored Jun 28, 2023
1 parent 6432aab commit 5b351e9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ggml-cuda.cu
Original file line number Diff line number Diff line change
Expand Up @@ -1292,7 +1292,7 @@ static __global__ void mul_mat_p021_f16_f32(const void * vx, const float * y, fl

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

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

Expand Down Expand Up @@ -1698,7 +1698,7 @@ static void ggml_mul_mat_vec_nc_f16_f32_cuda(
const dim3 block_nums(1, nrows_x, nchannels_x);
const dim3 block_dims(WARP_SIZE, 1, 1);
mul_mat_vec_nc_f16_f32<<<block_nums, block_dims, 0, stream>>>
(vx, y, dst, ncols_x, nrows_x, row_stride_x, nchannels_x, channel_stride_x);
(vx, y, dst, ncols_x, nrows_x, row_stride_x, channel_stride_x);
}

static void ggml_cpy_f32_f32_cuda(
Expand Down

0 comments on commit 5b351e9

Please sign in to comment.