AVX-512でクラッシュする問題を修正

[KazApps]

AVX-512のコードがクラッシュする原因としては、

• Accumulatorが64bytesでアライメントされていなかった。
• kSimdWidthを2倍し忘れていた。

の2点だったようで、そこを修正することで正常に動作するようになりました。 ただ、R9-7945HXではnpsの向上は誤差でした。このサイトにもある通りZEN4でのAVX-512に実装ではしょうがないのかもしれませんが。Xeonではもう少しnpsが向上するかもしれません。

[KazApps]

現在使われているNNUEのアーキテクチャには関係ないかもしれませんが、FeatureTransformer::Transformにおいて、AVX-512のコードのみ

for (IndexType i = 1; i < kRefreshTriggers.size(); ++i) {
    ...
}

がなくsumを正しく更新できてないような気がするので、修正しました(間違っていたらすみません)。

[yaneurao]

素晴らしいです！ありがとうございます！