improves env usage in tests

elstehle · elstehle · commit fd139665942a · 2025-11-18T12:07:54.000-08:00
diff --git a/cub/test/catch2_test_device_topk_keys.cu b/cub/test/catch2_test_device_topk_keys.cu
@@ -36,7 +36,7 @@ CUB_RUNTIME_FUNCTION static cudaError_t dispatch_topk_keys(
   NumOutItemsT k,
   cudaStream_t stream = 0)
 {
-  auto stream_env = cuda::std::execution::prop{cuda::get_stream_t{}, cuda::stream_ref{stream}};
+  auto stream_env = cuda::stream_ref{stream};
   auto requirements =
     cuda::execution::require(cuda::execution::determinism::not_guaranteed, cuda::execution::output_ordering::unsorted);
 
diff --git a/cub/test/catch2_test_device_topk_pairs.cu b/cub/test/catch2_test_device_topk_pairs.cu
@@ -37,7 +37,7 @@ CUB_RUNTIME_FUNCTION static cudaError_t dispatch_topk_pairs(
   NumOutItemsT k,
   cudaStream_t stream = 0)
 {
-  auto stream_env = cuda::std::execution::prop{cuda::get_stream_t{}, cuda::stream_ref{stream}};
+  auto stream_env = cuda::stream_ref{stream};
   auto requirements =
     cuda::execution::require(cuda::execution::determinism::not_guaranteed, cuda::execution::output_ordering::unsorted);
 

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ CUB_RUNTIME_FUNCTION static cudaError_t dispatch_topk_keys(`
`36`	`36`	`NumOutItemsT k,`
`37`	`37`	`cudaStream_t stream = 0)`
`38`	`38`	`{`
`39`		`- auto stream_env = cuda::std::execution::prop{cuda::get_stream_t{}, cuda::stream_ref{stream}};`
	`39`	`+ auto stream_env = cuda::stream_ref{stream};`
`40`	`40`	`auto requirements =`
`41`	`41`	`cuda::execution::require(cuda::execution::determinism::not_guaranteed, cuda::execution::output_ordering::unsorted);`
`42`	`42`
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ CUB_RUNTIME_FUNCTION static cudaError_t dispatch_topk_pairs(`
`37`	`37`	`NumOutItemsT k,`
`38`	`38`	`cudaStream_t stream = 0)`
`39`	`39`	`{`
`40`		`- auto stream_env = cuda::std::execution::prop{cuda::get_stream_t{}, cuda::stream_ref{stream}};`
	`40`	`+ auto stream_env = cuda::stream_ref{stream};`
`41`	`41`	`auto requirements =`
`42`	`42`	`cuda::execution::require(cuda::execution::determinism::not_guaranteed, cuda::execution::output_ordering::unsorted);`
`43`	`43`