diff options
Diffstat (limited to 'benchmarks')
-rw-r--r-- | benchmarks/benchmark.cc | 3 | ||||
-rw-r--r-- | benchmarks/benchmark_quantizer.cc | 2 | ||||
-rw-r--r-- | benchmarks/biasmultiply.cc | 2 |
3 files changed, 6 insertions, 1 deletions
diff --git a/benchmarks/benchmark.cc b/benchmarks/benchmark.cc index c6133bf..2dbe483 100644 --- a/benchmarks/benchmark.cc +++ b/benchmarks/benchmark.cc @@ -154,6 +154,7 @@ int main(int, char ** argv) { RunAll<sse2::Kernels16>(matrices, end, stats.sse2_16bit); } +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 std::cerr << "AVX2 8bit, 100 samples..." << std::endl; for (int samples = 0; samples < kSamples; ++samples) { RandomMatrices *end = (samples < 4) ? matrices_end : full_sample; @@ -165,7 +166,7 @@ int main(int, char ** argv) { RandomMatrices *end = (samples < 4) ? matrices_end : full_sample; RunAll<avx2::Kernels16>(matrices, end, stats.avx2_16bit); } - +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW std::cerr << "AVX512 8bit, 100 samples..." << std::endl; for (int samples = 0; samples < kSamples; ++samples) { diff --git a/benchmarks/benchmark_quantizer.cc b/benchmarks/benchmark_quantizer.cc index 5f36bd7..86d90dc 100644 --- a/benchmarks/benchmark_quantizer.cc +++ b/benchmarks/benchmark_quantizer.cc @@ -64,7 +64,9 @@ int main() { element = dist(gen); } QuantizerBench<intgemm::ssse3::Kernels8>(in.begin(), out.begin(), static_cast<intgemm::Index>(count)); +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 QuantizerBench<intgemm::avx2::Kernels8>(in.begin(), out.begin(), static_cast<intgemm::Index>(count)); +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW QuantizerBench<intgemm::avx512bw::Kernels8>(in.begin(), out.begin(), static_cast<intgemm::Index>(count)); #endif diff --git a/benchmarks/biasmultiply.cc b/benchmarks/biasmultiply.cc index 490bf3b..65deadb 100644 --- a/benchmarks/biasmultiply.cc +++ b/benchmarks/biasmultiply.cc @@ -161,6 +161,7 @@ int main(int argc, char ** argv) { std::cout << repeat << " iterations of Shifted SSSE3 took: " << newTimeSSSE3.count() << " seconds." << std::endl; +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 std::chrono::duration<double> oldAVX2_nobias = testOld_nobias<avx2::Kernels8>(1, 64, 8); for (int i = 0; i<repeat; i++) { oldAVX2_nobias += testOld_nobias<avx2::Kernels8>(8, 256, 256); @@ -196,6 +197,7 @@ int main(int argc, char ** argv) { } std::cout << repeat << " iterations of Shifted AVX2 took: " << newTimeAVX2.count() << " seconds." << std::endl; +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW if (kCPU < CPUType::AVX512BW) return 0; std::chrono::duration<double> oldAVX512_nobias = testOld_nobias<avx512bw::Kernels8>(1, 64, 8); |