Skip to content

brute_force_prefiltered test warp boundary but with less input - #2579

Open
robertmaynard wants to merge 3 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_brute_force_prefiltered_test_time
Open

brute_force_prefiltered test warp boundary but with less input#2579
robertmaynard wants to merge 3 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_brute_force_prefiltered_test_time

Conversation

@robertmaynard

Copy link
Copy Markdown
Contributor

For the longest running test items in brute_force_prefiltered.cu we can do a tweaks to improve runtime:

  • Use smaller input dim values which aren't a multiple of the vector width to still validate serial trailing loads
  • Increase the sparsity but keep about the kSddmmMaxSelectivity threshold

These combine together to remove 20 minutes from NEIGHBORS_TEST .

@robertmaynard
robertmaynard requested a review from a team as a code owner September 9, 2026 19:50
@robertmaynard robertmaynard added improvement Improves an existing functionality non-breaking Introduces a non-breaking change ci labels Sep 9, 2026

@tfeher tfeher left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks @robertmaynard!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants