diff --git a/CMakeLists.txt b/CMakeLists.txt index 68c707dc99abfd0d0831b8da2902c4cafd90b90a..90e915b0604efddc3a3c0c4b892579aaeb152dc1 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -153,6 +153,8 @@ set(KLEIDIAI_FILES_NEON_FP16_BF16 ) set(KLEIDIAI_FILES_NEON_ASM + kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c + kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qsi4c32pnrx8_qsu4c32s1s0_neon.c ) @@ -166,8 +168,6 @@ set(KLEIDIAI_FILES_NEON kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qsi4c32ps1s0scalef16_qsu4c32s16s0_neon.c kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qsi4cxps1s0_qsu4cxs1s0_neon.c kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qsi8cxp_qsi8cx_neon.c - kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c - kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c ) set(KLEIDIAI_FILES_NEON_DOTPROD_ASM @@ -420,6 +420,7 @@ if(KLEIDIAI_BUILD_TESTS) test/tests/bfloat16_test.cpp test/tests/float16_test.cpp test/tests/matmul_clamp_f32_qai8dxp_qsi4c32p_test.cpp + test/tests/matmul_clamp_f32_qsi8d32p_qai4c32p_test.cpp ) else() add_executable(kleidiai_test diff --git a/kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c b/kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c index 13571b59164f4b37b2b6780fa3bc1f13bdbc4608..1cfcaf71c727465834ee47a236cf2dbffecb3144 100644 --- a/kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c +++ b/kai/ukernels/matmul/pack/kai_lhs_quant_pack_qsi8d32pscalef32_f32_neon.c @@ -4,7 +4,7 @@ // SPDX-License-Identifier: Apache-2.0 // -#if !defined(__aarch64__) +#if !defined(__aarch64__) && !defined(_M_ARM64) #error This file must be compiled for AArch64. #else // Architectural features check. diff --git a/kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c b/kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c index b7183ac1bf6f02ec4192b2e610a2731f866ec126..e2ad2f5131fca36c35e6102be3809ef75b76aefe 100644 --- a/kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c +++ b/kai/ukernels/matmul/pack/kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.c @@ -4,7 +4,7 @@ // SPDX-License-Identifier: Apache-2.0 // -#if !defined(__aarch64__) +#if !defined(__aarch64__) && !defined(_M_ARM64) #error This file must be compiled for AArch64. #else // Architectural features check. #include "kai_rhs_pack_nxk_qai4c32p_qau4c32s0s1_f32_f32_f32_neon.h"