Skip to content

Commit

Permalink
Add kIsSingleSource to PairwiseDistanceEpilogueElementwise (#485)
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.

Also remove `raft_cutlass` from symbol exclusions.

Authors:
  - Kyle Edwards (https://github.com/KyleFromNVIDIA)

Approvers:
  - Bradley Dice (https://github.com/bdice)
  - Vyas Ramasubramani (https://github.com/vyasr)
  - Micka (https://github.com/lowener)

URL: #485
  • Loading branch information
KyleFromNVIDIA authored Nov 22, 2024
1 parent 06afd5b commit 710e9f5
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/pr.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ jobs:
with:
build_type: pull-request
enable_check_symbols: true
symbol_exclusions: (void (thrust::|cub::)|raft_cutlass)
symbol_exclusions: (void (thrust::|cub::))
conda-python-build:
needs: conda-cpp-build
secrets: inherit
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ jobs:
date: ${{ inputs.date }}
sha: ${{ inputs.sha }}
enable_check_symbols: true
symbol_exclusions: (void (thrust::|cub::)|raft_cutlass)
symbol_exclusions: (void (thrust::|cub::))
conda-cpp-tests:
secrets: inherit
uses: rapidsai/shared-workflows/.github/workflows/[email protected]
Expand Down
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 710e9f5

Please sign in to comment.