commit 0f349c496f82fa49aa09adbb257959eea4e74fd0
author Marat Dukhan <maratek@google.com> Wed Nov 27 11:58:54 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Nov 27 11:59:28 2019 -0800
tree a444fc9fb93b7a866c6728e2e9daf0ff0ebafd40
parent c72fa1e65f314612a9746a6ce783c0b218695f1b

AVX512F implementation of GEMM & IGEMM micro-kernels

PiperOrigin-RevId: 282810752

scripts/generate-f32-igemm.sh

diff --git a/scripts/generate-f32-igemm.sh b/scripts/generate-f32-igemm.sh
index 7161a27..fcc7c47 100755
--- a/scripts/generate-f32-igemm.sh
+++ b/scripts/generate-f32-igemm.sh
@@ -94,5 +94,14 @@
 tools/xngen src/f32-igemm/avx-broadcast.c.in -D MR=7 -D NR=8 -D FMA=3 -o src/f32-igemm/gen/7x8-fma3-broadcast.c
 tools/xngen src/f32-igemm/avx-broadcast.c.in -D MR=8 -D NR=8 -D FMA=3 -o src/f32-igemm/gen/8x8-fma3-broadcast.c
 
+################################# x86 AVX-512 #################################
+### AVX512F+BROADCAST micro-kernels
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=1 -D NR=16 -o src/f32-igemm/gen/1x16-avx512f-broadcast.c
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=4 -D NR=16 -o src/f32-igemm/gen/4x16-avx512f-broadcast.c
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=5 -D NR=16 -o src/f32-igemm/gen/5x16-avx512f-broadcast.c
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=6 -D NR=16 -o src/f32-igemm/gen/6x16-avx512f-broadcast.c
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=7 -D NR=16 -o src/f32-igemm/gen/7x16-avx512f-broadcast.c
+tools/xngen src/f32-igemm/avx512-broadcast.c.in -D MR=8 -D NR=16 -o src/f32-igemm/gen/8x16-avx512f-broadcast.c
+
 ################################## Unit tests #################################
 tools/generate-gemm-test.py --spec test/f32-igemm.yaml --output test/f32-igemm.cc