X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fsearch.cpp;h=f4f411c56d18f266f110d69ba5d4aa0b8c9707ad;hp=de6c2542d27805a25069290134107ba91583b282;hb=e215a88cddd16e09;hpb=e95e69515a8e30bcdcef84bd37656c3ea3099cd5;ds=sidebyside

diff --git a/src/search.cpp b/src/search.cpp
index de6c2542..f4f411c5 100644
--- a/src/search.cpp
+++ b/src/search.cpp
@@ -155,21 +155,17 @@ void Search::init() {
 
 size_t Search::perft(Position& pos, Depth depth) {
 
-  // At the last ply just return the number of legal moves (leaf nodes)
-  if (depth == ONE_PLY)
-      return MoveList<LEGAL>(pos).size();
-
   StateInfo st;
   size_t cnt = 0;
   CheckInfo ci(pos);
+  const bool leaf = depth == 2 * ONE_PLY;
 
   for (MoveList<LEGAL> it(pos); *it; ++it)
   {
       pos.do_move(*it, st, ci, pos.move_gives_check(*it, ci));
-      cnt += perft(pos, depth - ONE_PLY);
+      cnt += leaf ? MoveList<LEGAL>(pos).size() : perft(pos, depth - ONE_PLY);
       pos.undo_move(*it);
   }
-
   return cnt;
 }