Skip to content

Commit

Permalink
Add kIsSingleSource to PairwiseDistanceEpilogueElementwise
Browse files Browse the repository at this point in the history
With raft having recently migrated to cutlass 3.5.1, this field is
now required.
  • Loading branch information
KyleFromNVIDIA committed Nov 21, 2024
1 parent 06afd5b commit 41bf83e
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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_;
Expand Down

0 comments on commit 41bf83e

Please sign in to comment.