search: change nodes per second to float64

e0ff · Nov 26, 2023 · 402c2a5 · 402c2a5
1 parent 3d78df7
commit 402c2a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/cmd/rosaline/interfaces/uci.go b/cmd/rosaline/interfaces/uci.go
@@ -68,7 +68,7 @@ loop:
 						bestScore = results.Score
 					}
 
-					fmt.Printf("info depth %d score cp %d nodes %d nps %d time %d\n", depth, results.Score, results.Nodes, results.NPS, results.Time.Milliseconds())
+					fmt.Printf("info depth %d score cp %d nodes %d nps %f time %d\n", depth, results.Score, results.Nodes, results.NPS, results.Time.Milliseconds())
 				}
 
 				position.MakeMove(bestMove)

diff --git a/internal/search/negamax.go b/internal/search/negamax.go
@@ -22,7 +22,7 @@ type SearchResults struct {
 	Depth    int
 	Nodes    int
 	Time     time.Duration
-	NPS      int
+	NPS      float64
 }
 
 type NegamaxSearcher struct {
@@ -35,7 +35,6 @@ type NegamaxSearcher struct {
 	stop bool
 
 	nodes int
-	nps   int
 }
 
 func NewNegamaxSearcher(evaluator evaluation.Evaluator) NegamaxSearcher {
@@ -74,15 +73,15 @@ func (s NegamaxSearcher) Search(position chess.Position, depth int) SearchResult
 	}
 
 	elapsed := time.Since(start)
-	s.nps = int(int64(s.nodes) / elapsed.Milliseconds())
+	nps := float64(s.nodes) / float64(elapsed.Milliseconds())
 
 	return SearchResults{
 		BestMove: bestMove,
 		Score:    bestScore,
 		Depth:    depth,
 		Nodes:    s.nodes,
 		Time:     elapsed,
-		NPS:      s.nps,
+		NPS:      nps,
 	}
 }