diff options
Diffstat (limited to 'intgemm/intgemm.cc')
-rw-r--r-- | intgemm/intgemm.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/intgemm/intgemm.cc b/intgemm/intgemm.cc index dfb54f3..e5fe1b0 100644 --- a/intgemm/intgemm.cc +++ b/intgemm/intgemm.cc @@ -1,3 +1,14 @@ +#if defined(WASM) +// No header for CPUID since it's hard-coded. +#elif defined(__INTEL_COMPILER) +#include <immintrin.h> +#elif defined(_MSC_VER) +#include <intrin.h> +#else +// Assume GCC and clang style. +#include <cpuid.h> +#endif + #include "intgemm.h" #include "stats.h" |