runner: simplify tensor split parsing

d6af13ef · Michael Yang · a59f6652 · d6af13ef · d6af13ef
Commit d6af13ef authored Feb 26, 2025 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 10 deletions

runner/llamarunner/runner.go runner/llamarunner/runner.go +4 -5

runner/ollamarunner/runner.go runner/ollamarunner/runner.go +4 -5

No files found.
--- a/runner/llamarunner/runner.go
+++ b/runner/llamarunner/runner.go
@@ -943,12 +943,11 @@ func Execute(args []string) error {

 	var tensorSplitFloats []float32
 	if *tensorSplit != "" {
-		stringFloats := regexp.MustCompile(",").Split(*tensorSplit, -1)
-
-		tensorSplitFloats = make([]float32, 0, len(stringFloats))
-		for _, s := range stringFloats {
+		splits := strings.Split(*tensorSplit, ",")
+		tensorSplitFloats = make([]float32, len(splits))
+		for i, s := range splits {
 			f, _ := strconv.ParseFloat(s, 32)
-			tensorSplitFloats = append(tensorSplitFloats, float32(f))
+			tensorSplitFloats[i] = float32(f)
 		}
 	}


--- a/runner/ollamarunner/runner.go
+++ b/runner/ollamarunner/runner.go
@@ -881,12 +881,11 @@ func Execute(args []string) error {

 	var tensorSplitFloats []float32
 	if *tensorSplit != "" {
-		stringFloats := regexp.MustCompile(",").Split(*tensorSplit, -1)
-
-		tensorSplitFloats = make([]float32, 0, len(stringFloats))
-		for _, s := range stringFloats {
+		splits := strings.Split(*tensorSplit, ",")
+		tensorSplitFloats = make([]float32, len(splits))
+		for i, s := range splits {
 			f, _ := strconv.ParseFloat(s, 32)
-			tensorSplitFloats = append(tensorSplitFloats, float32(f))
+			tensorSplitFloats[i] = float32(f)
 		}
 	}