From 41bf83ebd4c6ba61c301ebcfe8c146258ecd5e75 Mon Sep 17 00:00:00 2001
From: Kyle Edwards <kyedwards@nvidia.com>
Date: Thu, 21 Nov 2024 12:15:35 -0500
Subject: [PATCH] Add kIsSingleSource to PairwiseDistanceEpilogueElementwise

With raft having recently migrated to cutlass 3.5.1, this field is
now required.
---
 cpp/src/distance/detail/pairwise_distance_epilogue_elementwise.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/src/distance/detail/pairwise_distance_epilogue_elementwise.h b/cpp/src/distance/detail/pairwise_distance_epilogue_elementwise.h
index f9955334d..f4a7feaba 100644
--- a/cpp/src/distance/detail/pairwise_distance_epilogue_elementwise.h
+++ b/cpp/src/distance/detail/pairwise_distance_epilogue_elementwise.h
@@ -61,6 +61,7 @@ class PairwiseDistanceEpilogueElementwise {
   using ElementT                      = ElementT_;
   static int const kElementsPerAccess = ElementsPerAccess;
   static int const kCount             = kElementsPerAccess;
+  static bool const kIsSingleSource   = true;
 
   using DistanceOp = DistanceOp_;
   using FinalOp    = FinalOp_;