diff options
author | Marat Dukhan <maratek@google.com> | 2020-08-10 11:37:23 -0700 |
---|---|---|
committer | XNNPACK Team <xnnpack-github-robot@google.com> | 2020-08-10 11:38:00 -0700 |
commit | bb00b1dda26b55c80f7a737ad374dfc312bd92b0 (patch) | |
tree | 260af38d3e6c650055de9d4b11d2e95779c26653 /bench/qs8-gemm.cc | |
parent | facecc526bf35d09ac3319f44072d640101b2b70 (diff) | |
download | XNNPACK-bb00b1dda26b55c80f7a737ad374dfc312bd92b0.tar.gz |
AVX512 variants of QS8 GEMM and IGEMM microkernels
PiperOrigin-RevId: 325850791
Diffstat (limited to 'bench/qs8-gemm.cc')
-rw-r--r-- | bench/qs8-gemm.cc | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/bench/qs8-gemm.cc b/bench/qs8-gemm.cc index 126bab2dd..0eb46de1a 100644 --- a/bench/qs8-gemm.cc +++ b/bench/qs8-gemm.cc @@ -318,6 +318,16 @@ static void GEMMBenchmark(benchmark::State& state, GEMMBenchmark(state, xnn_qs8_gemm_minmax_ukernel_3x8c8__avx2, 3, 8, 8, 1, benchmark::utils::CheckAVX2); } + static void qs8_gemm_2x16c8__avx512skx(benchmark::State& state, const char* net) { + GEMMBenchmark(state, xnn_qs8_gemm_minmax_ukernel_2x16c8__avx512skx, 2, 16, 8, 1, benchmark::utils::CheckAVX512SKX); + } + static void qs8_gemm_3x16c8__avx512skx(benchmark::State& state, const char* net) { + GEMMBenchmark(state, xnn_qs8_gemm_minmax_ukernel_3x16c8__avx512skx, 3, 16, 8, 1, benchmark::utils::CheckAVX512SKX); + } + static void qs8_gemm_4x16c8__avx512skx(benchmark::State& state, const char* net) { + GEMMBenchmark(state, xnn_qs8_gemm_minmax_ukernel_4x16c8__avx512skx, 4, 16, 8, 1, benchmark::utils::CheckAVX512SKX); + } + static void qs8_gemm_xw_2x8c8__avx2(benchmark::State& state, const char* net) { GEMMBenchmark(state, xnn_qs8_gemm_xw_minmax_ukernel_2x8c8__avx2, 2, 8, 8, 1, benchmark::utils::CheckAVX2); } @@ -325,6 +335,10 @@ static void GEMMBenchmark(benchmark::State& state, GEMMBenchmark(state, xnn_qs8_gemm_xw_minmax_ukernel_3x8c8__avx2, 3, 8, 8, 1, benchmark::utils::CheckAVX2); } + BENCHMARK_GEMM(qs8_gemm_2x16c8__avx512skx) + BENCHMARK_GEMM(qs8_gemm_3x16c8__avx512skx) + BENCHMARK_GEMM(qs8_gemm_4x16c8__avx512skx) + BENCHMARK_GEMM(qs8_gemm_2x8c8__avx2) BENCHMARK_GEMM(qs8_gemm_3x8c8__avx2) BENCHMARK_GEMM(qs8_gemm_xw_2x8c8__avx2) |