Fix a data race for NNUE