From 32f2daa80a24f16772e11d0afe7df7b6cb965b13 Mon Sep 17 00:00:00 2001
From: Mahesh Doijade <mdoijade@nvidia.com>
Date: Thu, 22 Jun 2023 04:46:15 -0700
Subject: [PATCH] pass correct stream to cutlass kernel launch, as it is not
 infered with initialization stream

---
 .../raft/distance/detail/pairwise_distance_cutlass_base.cuh     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/include/raft/distance/detail/pairwise_distance_cutlass_base.cuh b/cpp/include/raft/distance/detail/pairwise_distance_cutlass_base.cuh
index ccb3bd46bf..aeb862b06a 100644
--- a/cpp/include/raft/distance/detail/pairwise_distance_cutlass_base.cuh
+++ b/cpp/include/raft/distance/detail/pairwise_distance_cutlass_base.cuh
@@ -162,7 +162,7 @@ std::enable_if_t<ops::has_cutlass_op<OpT>::value> cutlassDistanceKernel(const Da
   RAFT_CUTLASS_TRY(cutlassDist_op.initialize(arguments, workspace.data(), stream));
 
   // Launch initialized CUTLASS kernel
-  RAFT_CUTLASS_TRY(cutlassDist_op());
+  RAFT_CUTLASS_TRY(cutlassDist_op(stream));
 }
 
 };  // namespace detail