Skip to content

Fix k-truss test failure on Blackwell#5495

Open
jnke2016 wants to merge 4 commits intorapidsai:mainfrom
jnke2016:fix_ktruss
Open

Fix k-truss test failure on Blackwell#5495
jnke2016 wants to merge 4 commits intorapidsai:mainfrom
jnke2016:fix_ktruss

Conversation

@jnke2016
Copy link
Copy Markdown
Contributor

The test_ktruss_subgraph_Graph test was failing on Blackwell (RTX PRO 6000) with nvcc 13.0 due to thrust::reduce_by_key silently corrupting its output when operating on zip_iterator<tuple<int32_t, int32_t>>. The bug is in nvcc 13.0 and is fixed in nvcc 13.2.

This PR:

  • Applies a workaround in k_truss_impl.cuh replacing the buggy thrust::reduce_by_key call with thrust::unique_copy + thrust::lower_bound / thrust::upper_bound
  • Re-enables the k-truss wheel tests that were previously skipped.

See #5494 for details. The original reduce_by_key call should be restored once nvcc 13.0 is no longer supported.

@jnke2016 jnke2016 requested review from a team as code owners April 16, 2026 19:44
@jnke2016 jnke2016 requested a review from AyodeAwe April 16, 2026 19:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant