pali112
/
ollama
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use top_k instead of argsort_top_k

This commit is contained in:
inforithmics 2026-01-01 23:57:34 +01:00
parent 18fdcc94e5
commit 3d00d68344
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ml/backend/ggml/ggml.go
View File

@ -1722,7 +1722,7 @@ func (t *Tensor) Duplicate(ctx ml.Context) ml.Tensor {
func (t *Tensor) TopK(ctx ml.Context, k int) ml.Tensor { func (t *Tensor) TopK(ctx ml.Context, k int) ml.Tensor {
return &Tensor{ return &Tensor{
b: t.b, 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)),
} }
} }