Enable NNUE pockets for gating variants

diff --git a/src/variant.h b/src/variant.h
index 81527ab..636197c 100644 (file)
--- a/src/variant.h
+++ b/src/variant.h
@@ -207,7 +207,7 @@ struct Variant {
                 : extinctionPieceTypes.find(COMMONER) != extinctionPieceTypes.end() ? COMMONER
                 : NO_PIECE_TYPE;
       nnueSquares = (maxRank + 1) * (maxFile + 1);
-      int nnuePockets = pieceDrops ? 2 * int(maxFile + 1) : 0;
+      int nnuePockets = pieceDrops || seirawanGating ? 2 * int(maxFile + 1) : 0;
       int nnueNonDropPieceIndices = (2 * pieceTypes.size() - 1) * nnueSquares;
       nnuePieceIndices = nnueNonDropPieceIndices + 2 * (pieceTypes.size() - 1) * nnuePockets;
       int i = 0;