From 6a2197328065fe38ac62d9e76e3e943fd297e73c Mon Sep 17 00:00:00 2001 From: tslil Date: Sun, 20 Sep 2026 08:28:09 +0100 Subject: try combining loops --- include/nn.c | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'include/nn.c') diff --git a/include/nn.c b/include/nn.c index 9f15ec2..926fcf7 100644 --- a/include/nn.c +++ b/include/nn.c @@ -127,11 +127,12 @@ float eval_net(const float *board, const unsigned int n_inp, const float *w, for (unsigned int d1 = 0; d1 < n_dense; d1++) dense1[d1] += b * wrow[d1]; } - for (unsigned int d1 = 0; d1 < n_dense; d1++) - dense1[d1] = fast_silu(dense1[d1]); float z = logit_b; - for (unsigned int d1 = 0; d1 < n_dense; d1++) + for (unsigned int d1 = 0; d1 < n_dense; d1++) { + const float h = dense1[d1]; + dense1[d1] = fast_silu(h); z += dense1[d1] * logit_w[d1]; + } return parity * fast_tanh(z); } -- cgit v1.3.1