Skip to content

Commit 0a737d2

Browse files
committed
remove unused parameter
ggml-ci
1 parent 35386e8 commit 0a737d2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-cuda/argmax.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
#include "common.cuh"
66
#include "sum.cuh"
77

8-
static __global__ void argmax_f32(const float * x, int32_t * dst, const int64_t ncols, const int64_t nrows) {
8+
static __global__ void argmax_f32(const float * x, int32_t * dst, const int64_t ncols) {
99
const int64_t row = blockIdx.x;
1010

1111
float maxval = -FLT_MAX;
@@ -85,5 +85,5 @@ void ggml_cuda_argmax(ggml_backend_cuda_context & ctx, ggml_tensor * dst) {
8585
const dim3 blocks_dim(std::min<int64_t>(ne00, 1024), 1, 1);
8686
const dim3 blocks_num(num_blocks, 1, 1);
8787

88-
argmax_f32<<<blocks_num, blocks_dim, 0, stream>>>(src0_d, dst_d, ne00, nrows);
88+
argmax_f32<<<blocks_num, blocks_dim, 0, stream>>>(src0_d, dst_d, ne00);
8989
}

0 commit comments

Comments
 (0)