]> git.sesse.net Git - stockfish/blob - src/nnue/nnue_architecture.h
Add NNUE evaluation
[stockfish] / src / nnue / nnue_architecture.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 // Input features and network structure used in NNUE evaluation function
20
21 #ifndef NNUE_ARCHITECTURE_H_INCLUDED
22 #define NNUE_ARCHITECTURE_H_INCLUDED
23
24 // Defines the network structure
25 #include "architectures/halfkp_256x2-32-32.h"
26
27 namespace Eval::NNUE {
28
29   static_assert(kTransformedFeatureDimensions % kMaxSimdWidth == 0, "");
30   static_assert(Network::kOutputDimensions == 1, "");
31   static_assert(std::is_same<Network::OutputType, std::int32_t>::value, "");
32
33   // Trigger for full calculation instead of difference calculation
34   constexpr auto kRefreshTriggers = RawFeatures::kRefreshTriggers;
35
36 }  // namespace Eval::NNUE
37
38 #endif // #ifndef NNUE_ARCHITECTURE_H_INCLUDED