Is this a duplicate?

• I confirmed there appear to be no duplicate issues for this request and that I agree to the Code of Conduct

Area

CUB

Is your feature request related to a problem? Please describe.

Multiple CUB routines fail if executed in a CUDA graph

Describe the solution you'd like

Add test cases in the unit test to ensure that all routines support CUDA graph capture

Describe alternatives you've considered

No response

Additional context

No response