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

Merge 3d00d68344 into 76912c062a

This commit is contained in:
Thomas Stocker 2026-01-06 15:45:05 +08:00 committed by GitHub
parent 76912c062a 3d00d68344
commit e611ec3429
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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)),
} }
} }