aboutsummaryrefslogtreecommitdiff
path: root/include/nn.h
diff options
context:
space:
mode:
authortslil <tslil@posteo.de>2026-09-08 11:55:28 +0100
committertslil <tslil@posteo.de>2026-09-09 21:45:17 +0100
commit1c9e266ae25f7f6067806d139e5971ce8513d2f6 (patch)
tree9b935954310a56ee7fdba07db5f03888fc5af7b0 /include/nn.h
parent7ef68374e8a58684b299b0f46f7ed850d5a87190 (diff)
~25% speedup in selfplay due to: incremental zobrist hashing, smarter road checking, re-using the input computed for the NN
Diffstat (limited to 'include/nn.h')
-rw-r--r--include/nn.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/include/nn.h b/include/nn.h
index 9e576c1..871bd0f 100644
--- a/include/nn.h
+++ b/include/nn.h
@@ -35,6 +35,8 @@ void nn_prepare_head(float out_head[3], tak_state_p state);
void nn_prepare_input(float *cur_board, tak_state_p state,
const float embedding[NN_N_EMB]);
+
+int nn_last_cell_idxs(uint8_t *out, tak_state_p state);
// ===================================================================
// Network
// ===================================================================