From 04ca98b82f37e3dc6b43014a8fe08e7bb5db3030 Mon Sep 17 00:00:00 2001
From: Allard Hendriksen <ahendriksen@nvidia.com>
Date: Wed, 5 Oct 2022 16:17:56 +0200
Subject: [PATCH] pairwise_distance_base: Fix typo

This results in subtle issues with non-square KernelPolicy, as found in
fusedL2KNN.
---
 cpp/include/raft/distance/detail/pairwise_distance_base.cuh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/include/raft/distance/detail/pairwise_distance_base.cuh b/cpp/include/raft/distance/detail/pairwise_distance_base.cuh
index a2dffad808..b28c3a3de4 100644
--- a/cpp/include/raft/distance/detail/pairwise_distance_base.cuh
+++ b/cpp/include/raft/distance/detail/pairwise_distance_base.cuh
@@ -122,8 +122,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)
   {