Skip to content

Conversation

sudhu2k
Copy link

@sudhu2k sudhu2k commented Jul 25, 2025

Summary

Future proofing by changing the WARP_THREADS macro definition to a rocprim::arch::wavefront::max_size(), since rocprim::warp_size()` function will be removed or deprecated in future versions of ROCm.

What's Changed

  • In csrc/selective_scan/reverse_scan.cuh, the definition of WARP_THREADS has been updated for the ROCm build path.
  • The change modifies the macro to use rocprim::arch::wavefront::max_size(), which correctly resolves the warp size.

Why It's Important

The rocprim::warp_size() function will be removed or deprecated in future versions

How to Test

This is a build-time fix for future ROCm versions

@sudhu2k sudhu2k force-pushed the hipcub_warp_threads_deprecation branch from 9ed6ba1 to 673c80b Compare July 25, 2025 17:14
@sudhu2k
Copy link
Author

sudhu2k commented Jul 30, 2025

Hi @tridao @albertfgu,
Can some maintainers please take a look at this PR, it'll be of great help! Thanks again!

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