mirror of
https://github.com/HChaZZY/Stockfish.git
synced 2025-12-25 03:26:24 +08:00
8ee9905d8b
Passed STC: LLR: 2.94 (-2.94,2.94) <-1.75,0.25> Total: 94208 W: 24270 L: 24112 D: 45826 Ptnml(0-2): 286, 11186, 24009, 11330, 293 https://tests.stockfishchess.org/tests/view/6635ddd773559a8aa8582826 Passed LTC: LLR: 2.95 (-2.94,2.94) <-1.75,0.25> Total: 114960 W: 29107 L: 28982 D: 56871 Ptnml(0-2): 37, 12683, 31924, 12790, 46 https://tests.stockfishchess.org/tests/view/663604a973559a8aa85881ed closes #5214 Bench 1653939
127 lines
4.0 KiB
C++
127 lines
4.0 KiB
C++
/*
|
|
Stockfish, a UCI chess playing engine derived from Glaurung 2.1
|
|
Copyright (C) 2004-2024 The Stockfish developers (see AUTHORS file)
|
|
|
|
Stockfish is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
Stockfish is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef NETWORK_H_INCLUDED
|
|
#define NETWORK_H_INCLUDED
|
|
|
|
#include <cstdint>
|
|
#include <iostream>
|
|
#include <optional>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
#include "../misc.h"
|
|
#include "../position.h"
|
|
#include "../types.h"
|
|
#include "nnue_architecture.h"
|
|
#include "nnue_feature_transformer.h"
|
|
#include "nnue_misc.h"
|
|
#include "nnue_accumulator.h"
|
|
|
|
namespace Stockfish::Eval::NNUE {
|
|
|
|
enum class EmbeddedNNUEType {
|
|
BIG,
|
|
SMALL,
|
|
};
|
|
|
|
|
|
template<typename Arch, typename Transformer>
|
|
class Network {
|
|
static constexpr IndexType FTDimensions = Arch::TransformedFeatureDimensions;
|
|
|
|
public:
|
|
Network(EvalFile file, EmbeddedNNUEType type) :
|
|
evalFile(file),
|
|
embeddedType(type) {}
|
|
|
|
void load(const std::string& rootDirectory, std::string evalfilePath);
|
|
bool save(const std::optional<std::string>& filename) const;
|
|
|
|
Value evaluate(const Position& pos,
|
|
AccumulatorCaches::Cache<FTDimensions>* cache,
|
|
bool adjusted = false,
|
|
int* complexity = nullptr) const;
|
|
|
|
|
|
void hint_common_access(const Position& pos,
|
|
AccumulatorCaches::Cache<FTDimensions>* cache) const;
|
|
|
|
void verify(std::string evalfilePath) const;
|
|
NnueEvalTrace trace_evaluate(const Position& pos,
|
|
AccumulatorCaches::Cache<FTDimensions>* cache) const;
|
|
|
|
private:
|
|
void load_user_net(const std::string&, const std::string&);
|
|
void load_internal();
|
|
|
|
void initialize();
|
|
|
|
bool save(std::ostream&, const std::string&, const std::string&) const;
|
|
std::optional<std::string> load(std::istream&);
|
|
|
|
bool read_header(std::istream&, std::uint32_t*, std::string*) const;
|
|
bool write_header(std::ostream&, std::uint32_t, const std::string&) const;
|
|
|
|
bool read_parameters(std::istream&, std::string&) const;
|
|
bool write_parameters(std::ostream&, const std::string&) const;
|
|
|
|
// Input feature converter
|
|
LargePagePtr<Transformer> featureTransformer;
|
|
|
|
// Evaluation function
|
|
AlignedPtr<Arch> network[LayerStacks];
|
|
|
|
EvalFile evalFile;
|
|
EmbeddedNNUEType embeddedType;
|
|
|
|
// Hash value of evaluation function structure
|
|
static constexpr std::uint32_t hash = Transformer::get_hash_value() ^ Arch::get_hash_value();
|
|
|
|
template<IndexType Size>
|
|
friend struct AccumulatorCaches::Cache;
|
|
};
|
|
|
|
// Definitions of the network types
|
|
using SmallFeatureTransformer =
|
|
FeatureTransformer<TransformedFeatureDimensionsSmall, &StateInfo::accumulatorSmall>;
|
|
using SmallNetworkArchitecture =
|
|
NetworkArchitecture<TransformedFeatureDimensionsSmall, L2Small, L3Small>;
|
|
|
|
using BigFeatureTransformer =
|
|
FeatureTransformer<TransformedFeatureDimensionsBig, &StateInfo::accumulatorBig>;
|
|
using BigNetworkArchitecture = NetworkArchitecture<TransformedFeatureDimensionsBig, L2Big, L3Big>;
|
|
|
|
using NetworkBig = Network<BigNetworkArchitecture, BigFeatureTransformer>;
|
|
using NetworkSmall = Network<SmallNetworkArchitecture, SmallFeatureTransformer>;
|
|
|
|
|
|
struct Networks {
|
|
Networks(NetworkBig&& nB, NetworkSmall&& nS) :
|
|
big(std::move(nB)),
|
|
small(std::move(nS)) {}
|
|
|
|
NetworkBig big;
|
|
NetworkSmall small;
|
|
};
|
|
|
|
|
|
} // namespace Stockfish
|
|
|
|
#endif
|