Merge 3d00d68344 into 76912c062a

2026-01-06 15:45:05 +08:00 · 2026-01-06 15:45:05 +08:00 · e611ec3429
parent 76912c062a 3d00d68344
commit e611ec3429
1 changed files with 1 additions and 1 deletions
--- 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)),
 	}
 }