pairwise_distance_base: Fix typo

This results in subtle issues with non-square KernelPolicy, as found in fusedL2KNN.
rapidsai · Oct 5, 2022 · 9029eed · 9029eed
1 parent b5aef79
commit 9029eed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/include/raft/distance/detail/pairwise_distance_base.cuh b/cpp/include/raft/distance/detail/pairwise_distance_base.cuh
@@ -121,8 +121,8 @@ struct PairwiseDistances : public BaseClass {
       epilog_op(_epilog_op),
       fin_op(_fin_op),
       rowEpilog_op(_rowEpilog_op),
-      grid_stride_m(P::Nblk * gridDim.y),
-      grid_stride_n(P::Mblk * gridDim.x),
+      grid_stride_m(P::Mblk * gridDim.y),
+      grid_stride_n(P::Nblk * gridDim.x),
       grid_offset_m(P::Mblk * blockIdx.y),
       grid_offset_n(P::Nblk * blockIdx.x)
   {