diff --git a/BUILD.bazel b/BUILD.bazel index 944b35e72f9b..7c182f52722e 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -978,7 +978,7 @@ xnnpack_microkernel_cc_library( gcc_copts = xnnpack_gcc_std_copts(), msvc_copts = xnnpack_msvc_std_copts(), deps = MICROKERNEL_DEPS + xnnpack_if_kleidiai_enabled([ - "@KleidiAI//kai/ukernels/matmul", + "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp1x8_qsi4cxp4x8_1x4x32_neon_dotprod", "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp1x8_qsi4cxp8x8_1x8x32_neon_dotprod", ]), ) @@ -1002,7 +1002,9 @@ xnnpack_microkernel_cc_library( gcc_copts = xnnpack_gcc_std_copts(), msvc_copts = xnnpack_msvc_std_copts(), deps = MICROKERNEL_DEPS + xnnpack_if_kleidiai_enabled([ - "@KleidiAI//kai/ukernels/matmul", + "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp4x8_qsi4cxp4x8_4x4x32_neon_i8mm", + "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp4x8_qsi4cxp4x8_8x4x32_neon_i8mm", + "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp4x8_qsi4cxp8x8_4x8x32_neon_i8mm", "@KleidiAI//kai/ukernels/matmul:clamp_f32_qai8dxp4x8_qsi4cxp8x8_8x8x32_neon_i8mm", ]), )