From 3d00d68344f233bdafb79eea4bab2c5692914fd8 Mon Sep 17 00:00:00 2001 From: inforithmics Date: Thu, 1 Jan 2026 23:57:34 +0100 Subject: [PATCH] Use top_k instead of argsort_top_k --- ml/backend/ggml/ggml.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml/backend/ggml/ggml.go b/ml/backend/ggml/ggml.go index ebcc1d86f..9826e14c6 100644 --- a/ml/backend/ggml/ggml.go +++ b/ml/backend/ggml/ggml.go @@ -1722,7 +1722,7 @@ func (t *Tensor) Duplicate(ctx ml.Context) ml.Tensor { func (t *Tensor) TopK(ctx ml.Context, k int) ml.Tensor { return &Tensor{ b: t.b, - t: C.ggml_argsort_top_k(ctx.(*Context).ctx, t.t, C.int(k)), + t: C.ggml_top_k(ctx.(*Context).ctx, t.t, C.int(k)), } }