]> git.sesse.net Git - stockfish/commitdiff
Fix export of the feature transformer.
authorTomasz Sobczyk <tomasz.sobczyk1997@gmail.com>
Sat, 29 May 2021 09:40:40 +0000 (11:40 +0200)
committerJoost VandeVondele <Joost.VandeVondele@gmail.com>
Sun, 30 May 2021 19:31:58 +0000 (21:31 +0200)
PSQT export was missing.

fixes #3507

closes https://github.com/official-stockfish/Stockfish/pull/3508

No functional change

src/nnue/nnue_feature_transformer.h

index 741d97cf1190c539bdcf2766ae6852bdefc52a12..c249d3e70184edd46be1fef51fa5fe874d16fcd0 100644 (file)
@@ -165,6 +165,8 @@ namespace Stockfish::Eval::NNUE {
         write_little_endian<BiasType>(stream, biases[i]);
       for (std::size_t i = 0; i < HalfDimensions * InputDimensions; ++i)
         write_little_endian<WeightType>(stream, weights[i]);
+      for (std::size_t i = 0; i < PSQTBuckets * InputDimensions; ++i)
+        write_little_endian<PSQTWeightType>(stream, psqtWeights[i]);
       return !stream.fail();
     }