diff options
Diffstat (limited to 'intgemm/kernels.h')
-rw-r--r-- | intgemm/kernels.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/intgemm/kernels.h b/intgemm/kernels.h new file mode 100644 index 0000000..ee35966 --- /dev/null +++ b/intgemm/kernels.h @@ -0,0 +1,24 @@ +#pragma once + +#include "intgemm/intgemm_config.h" +#include "intrinsics.h" +#include "types.h" +#include "utils.h" +#include "vec_traits.h" + +#include <cstdlib> + +#define KERNELS_THIS_IS_SSE2 +#include "kernels/implementations.inl" +#undef KERNELS_THIS_IS_SSE2 + +#define KERNELS_THIS_IS_AVX2 +#include "kernels/implementations.inl" +#undef KERNELS_THIS_IS_AVX2 + +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW +#define KERNELS_THIS_IS_AVX512BW +#include "kernels/implementations.inl" +#undef KERNELS_THIS_IS_AVX512BW +#endif + |