diff options
Diffstat (limited to 'js/src/intgemm/IntegerGemmIntrinsic.cpp')
| -rw-r--r-- | js/src/intgemm/IntegerGemmIntrinsic.cpp | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/js/src/intgemm/IntegerGemmIntrinsic.cpp b/js/src/intgemm/IntegerGemmIntrinsic.cpp index aebb0f5f02..0e79ccad36 100644 --- a/js/src/intgemm/IntegerGemmIntrinsic.cpp +++ b/js/src/intgemm/IntegerGemmIntrinsic.cpp @@ -22,10 +22,9 @@ #if defined(USE_AVX512BW) # if defined(USE_AVX512VNNI) -# define SUPPORTED_ARCHS \ - xsimd::arch_list<xsimd::avx512vnni<xsimd::avx512bw>, \ - xsimd::avx512bw, xsimd::avx2, \ - xsimd::ssse3, xsimd::sse2> +# define SUPPORTED_ARCHS \ + xsimd::arch_list<xsimd::avx512vnni<xsimd::avx512bw>, xsimd::avx512bw, \ + xsimd::avx2, xsimd::ssse3, xsimd::sse2> # elif defined(USE_AVXVNNI) # define SUPPORTED_ARCHS \ xsimd::arch_list<xsimd::avx512bw, xsimd::avxvnni, xsimd::avx2, \ @@ -45,7 +44,12 @@ #elif defined(USE_SSE2) # define SUPPORTED_ARCHS xsimd::arch_list<xsimd::sse2> #elif defined(USE_NEON) and defined(XSIMD_WITH_NEON64) -# define SUPPORTED_ARCHS xsimd::arch_list<xsimd::neon64> +# if defined(USE_NEON_I8MM) +# define SUPPORTED_ARCHS \ + xsimd::arch_list<xsimd::i8mm<xsimd::neon64>, xsimd::neon64> +# else +# define SUPPORTED_ARCHS xsimd::arch_list<xsimd::neon64> +# endif #else # error no supported architecture #endif |