git.sesse.net Git - stockfish/blob - src/nnue/nnue_accumulator.h

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2020 The Stockfish developers (see AUTHORS file)
   4
   5   Stockfish is free software: you can redistribute it and/or modify
   6   it under the terms of the GNU General Public License as published by
   7   the Free Software Foundation, either version 3 of the License, or
   8   (at your option) any later version.
   9
  10   Stockfish is distributed in the hope that it will be useful,
  11   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13   GNU General Public License for more details.
  14
  15   You should have received a copy of the GNU General Public License
  16   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 // Class for difference calculation of NNUE evaluation function
  20
  21 #ifndef NNUE_ACCUMULATOR_H_INCLUDED
  22 #define NNUE_ACCUMULATOR_H_INCLUDED
  23
  24 #include "nnue_architecture.h"
  25
  26 namespace Eval::NNUE {
  27
  28   // The accumulator of a StateInfo without parent is set to the INIT state
  29   enum AccumulatorState { EMPTY, COMPUTED, INIT };
  30
  31   // Class that holds the result of affine transformation of input features
  32   struct alignas(kCacheLineSize) Accumulator {
  33     std::int16_t
  34         accumulation[2][kRefreshTriggers.size()][kTransformedFeatureDimensions];
  35     AccumulatorState state[2];
  36   };
  37
  38 }  // namespace Eval::NNUE
  39
  40 #endif // NNUE_ACCUMULATOR_H_INCLUDED