Fix issue with incorrect stride.

PiperOrigin-RevId: 662423833
google · Aug 13, 2024 · ddeb3f7 · ddeb3f7
1 parent 15a7a11
commit ddeb3f7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/packing.c b/src/packing.c
@@ -1623,11 +1623,10 @@ void xnn_pack_qu8_weights_and_biases(
 size_t xnn_packed_stride_kai_qs4_weights_and_biases(
     const struct xnn_gemm_config* gemm_config, size_t k, size_t unused_k_stride,
     size_t extra_bytes) {
-  const uint32_t nr = gemm_config->nr;
   const uint32_t kr = UINT32_C(1) << gemm_config->log2_kr;
   const uint32_t sr = UINT32_C(1) << gemm_config->log2_sr;
-  return kai_get_rhs_packed_stride_rhs_pack_kxn_qsi4cxp_qsu4cxs1s0(k, nr, kr,
-                                                                   sr);
+  return kai_get_rhs_packed_stride_rhs_pack_kxn_qsi4cxp_qsu4cxs1s0(k, /*nr=*/1,
+                                                                   kr, sr);
 }
 
 void xnn_pack_kai_qs4_weights_and_biases(