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

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2023 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 // header used in NNUE evaluation function
  20
  21 #ifndef NNUE_EVALUATE_NNUE_H_INCLUDED
  22 #define NNUE_EVALUATE_NNUE_H_INCLUDED
  23
  24 #include "nnue_feature_transformer.h"
  25
  26 #include <memory>
  27
  28 namespace Stockfish::Eval::NNUE {
  29
  30   // Hash value of evaluation function structure
  31   constexpr std::uint32_t HashValue =
  32       FeatureTransformer::get_hash_value() ^ Network::get_hash_value();
  33
  34
  35   // Deleter for automating release of memory area
  36   template <typename T>
  37   struct AlignedDeleter {
  38     void operator()(T* ptr) const {
  39       ptr->~T();
  40       std_aligned_free(ptr);
  41     }
  42   };
  43
  44   template <typename T>
  45   struct LargePageDeleter {
  46     void operator()(T* ptr) const {
  47       ptr->~T();
  48       aligned_large_pages_free(ptr);
  49     }
  50   };
  51
  52   template <typename T>
  53   using AlignedPtr = std::unique_ptr<T, AlignedDeleter<T>>;
  54
  55   template <typename T>
  56   using LargePagePtr = std::unique_ptr<T, LargePageDeleter<T>>;
  57
  58   std::string trace(Position& pos);
  59   Value evaluate(const Position& pos, bool adjusted = false, int* complexity = nullptr);
  60   void hint_common_parent_position(const Position& pos);
  61
  62   bool load_eval(std::string name, std::istream& stream);
  63   bool save_eval(std::ostream& stream);
  64   bool save_eval(const std::optional<std::string>& filename);
  65
  66 }  // namespace Stockfish::Eval::NNUE
  67
  68 #endif // #ifndef NNUE_EVALUATE_NNUE_H_INCLUDED