Set XNN_MAX_MR to 16 (was 32)

- Change XNN_MAX_MR to 16 except for aarch64 with KLEIDIAI PiperOrigin-RevId: 704794505
google · Dec 10, 2024 · d0e590e · d0e590e
1 parent a536778
commit d0e590e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/xnnpack/microfnptr.h b/src/xnnpack/microfnptr.h
@@ -2313,8 +2313,11 @@ struct xnn_hmp_qp8gemm_bl_ukernel {
 };
 
 // Largest GEMM/IGEMM MR used in init.c is 16 (x86 AVX512AMX).
-// Largest GEMM/IGEMM MR is 8 in e2e benchmarks.
+#if XNN_ARCH_ARM64 && XNN_ENABLE_KLEIDIAI
 #define XNN_MAX_MR 32
+#else
+#define XNN_MAX_MR 16
+#endif
 
 struct gemm_fused_ukernels {
   union {