Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/intgemm/intgemm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'benchmarks/benchmark.cc')
-rw-r--r--benchmarks/benchmark.cc6
1 files changed, 3 insertions, 3 deletions
diff --git a/benchmarks/benchmark.cc b/benchmarks/benchmark.cc
index 26b0ac5..6063d5c 100644
--- a/benchmarks/benchmark.cc
+++ b/benchmarks/benchmark.cc
@@ -196,7 +196,7 @@ int main(int, char ** argv) {
RunAll<AVX2_16bit>(matrices, end, stats.avx2_16bit);
}
-#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512
+#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW
std::cerr << "AVX512 8bit, 100 samples..." << std::endl;
for (int samples = 0; samples < kSamples; ++samples) {
RandomMatrices *end = (samples < 4) ? matrices_end : full_sample;
@@ -225,7 +225,7 @@ int main(int, char ** argv) {
std::cout << "Multiply\t" << matrices[i].A_rows << '\t' << matrices[i].width << '\t' << matrices[i].B_cols << '\t' << "Samples=" << (kOutlierThreshold * stats.sse2_16bit[i].size()) << '\n';
Print<SSSE3_8bit>(stats.ssse3_8bit, i);
Print<AVX2_8bit>(stats.avx2_8bit, i);
-#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512
+#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW
Print<AVX512_8bit>(stats.avx512_8bit, i);
#endif
#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512VNNI
@@ -233,7 +233,7 @@ int main(int, char ** argv) {
#endif
Print<SSE2_16bit>(stats.sse2_16bit, i);
Print<AVX2_16bit>(stats.avx2_16bit, i);
-#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512
+#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW
Print<AVX512_16bit>(stats.avx512_16bit, i);
#endif
}