Skip to content

Reduce NEIGHBORS_ALL_NEIGHBORS_TEST combinatorial test space - #2573

Open
robertmaynard wants to merge 2 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_test_time
Open

Reduce NEIGHBORS_ALL_NEIGHBORS_TEST combinatorial test space#2573
robertmaynard wants to merge 2 commits into
NVIDIA:mainfrom
robertmaynard:fea/improve_neighbors_test_time

Conversation

@robertmaynard

Copy link
Copy Markdown
Contributor

Moves NEIGHBORS_ALL_NEIGHBORS_TEST runtime from 34 minutes to 4 minutes by making the combinatorial space a little sparse.

Primarily this is done by doing:

  1. Data locality/movement testing doesn't need to happen across all n_row * dim * graph_degree inputs

  2. Make n_row * dim * graph_degree sparser for batched tests. This is done by testing min, max and ~diagonal for this test space.

Moves NEIGHBORS_ALL_NEIGHBORS_TEST runtime from 34 minutes
to 4 minutes by making the combinatorial space a little sparse.

Primarily this is done by doing:

1. Data locality/movement testing doesn't need to happen across
all n_row * dim * graph_degree inputs

2. Make n_row * dim * graph_degree sparser for batched tests. This
is done by testing min, max and ~diagonal for this test space.
@robertmaynard
robertmaynard requested a review from a team as a code owner September 9, 2026 15:28
@robertmaynard robertmaynard added feature request New feature or request non-breaking Introduces a non-breaking change ci labels Sep 9, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci feature request New feature or request non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant