highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-11-10 22:07:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: using testing.B.Loop

Browse Source 
Signed-off-by: yajianggroup <yajianggroup@outlook.com>
This commit is contained in:
yajianggroup
2025-09-23 16:05:59 +08:00
committed by Michael Yang
parent 3d32249c74
commit df411c4b02
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model/bytepairencoding_test.go
View File

@@ -251,7 +251,7 @@ func BenchmarkBytePairEncoding(b *testing.B) {
bts := bts[:n] bts := bts[:n]
b.Run("encode"+strconv.Itoa(n), func(b *testing.B) { b.Run("encode"+strconv.Itoa(n), func(b *testing.B) {
b.ResetTimer() b.ResetTimer()
for range b.N { for b.Loop() {
_, err := tokenizer.Encode(string(bts), true) _, err := tokenizer.Encode(string(bts), true)
if err != nil { if err != nil {
b.Fatal(err) b.Fatal(err)
@@ -266,7 +266,7 @@ func BenchmarkBytePairEncoding(b *testing.B) {
} }
b.ResetTimer() b.ResetTimer()
for range b.N { for b.Loop() {
_, err := tokenizer.Decode(ids) _, err := tokenizer.Decode(ids)
if err != nil { if err != nil {
b.Fatal(err) b.Fatal(err)
@@ -276,7 +276,7 @@ func BenchmarkBytePairEncoding(b *testing.B) {
b.Run("split"+strconv.Itoa(n), func(b *testing.B) { b.Run("split"+strconv.Itoa(n), func(b *testing.B) {
b.ResetTimer() b.ResetTimer()
for range b.N { for b.Loop() {
slices.Collect(tokenizer.split(string(bts))) slices.Collect(tokenizer.split(string(bts)))
} }
}) })