diff --git a/kai/ukernels/matmul/BUILD.bazel b/kai/ukernels/matmul/BUILD.bazel index 96c267801d3af7ca11d536bc4d1f89757bfc3004..67aa0d9ec8a9748f6c9d6a939541a72d7fbbc3b0 100644 --- a/kai/ukernels/matmul/BUILD.bazel +++ b/kai/ukernels/matmul/BUILD.bazel @@ -118,6 +118,13 @@ kai_c_library( cpu_uarch = kai_cpu_neon(), ) +kai_c_library( + name = "lhs_quant_pack_qsi8d32p_f32", + srcs = ["pack/kai_lhs_quant_pack_qsi8d32p_f32.c"], + hdrs = ["pack/kai_lhs_quant_pack_qsi8d32p_f32.h"], + cpu_uarch = kai_cpu_neon(), +) + kai_c_library( name = "lhs_pack_f32p2vlx1_f32_sme", srcs = ["pack/kai_lhs_pack_f32p2vlx1_f32_sme.c"], @@ -284,6 +291,7 @@ kai_c_library( ":clamp_f32_qsi8d32p_qsi4c32p_interface", ":lhs_pack_f32p2vlx1_f32_sme", ":lhs_quant_pack_qai8dxp_f32", + ":lhs_quant_pack_qsi8d32p_f32", ":rhs_pack_kxn_f16p16x1biasf16_f16_f16_neon", ":rhs_pack_kxn_f32p2vlx1biasf32_f32_f32_sme", ":rhs_pack_kxn_f32pbiasf32_f32_f32_neon", diff --git a/test/BUILD.bazel b/test/BUILD.bazel index c806b5d54fdd54ddd92082dfcf502ffbb2c3d29a..94838c50536867806664f706da2730b2689dc8b7 100644 --- a/test/BUILD.bazel +++ b/test/BUILD.bazel @@ -52,11 +52,7 @@ kai_cxx_library( cc_test( name = "kleidiai_test", - srcs = [ - "tests/matmul_clamp_f32_qai8dxp_qsi4c32p_test.cpp", - "tests/matmul_clamp_f32_qai8dxp_qsi4cxp_test.cpp", - "tests/matmul_test.cpp", - ], + srcs = glob(["tests/*.cpp"]), copts = kai_cxxopts(kai_cpu_bf16() + kai_cpu_fp16()), includes = [], linkstatic = True,