| 926 | |
| 927 | template<bool UseGraphs = false> |
| 928 | void testLargeIndicesInclusiveScanByKey() |
| 929 | { |
| 930 | auto inclusive_scan_by_key = [](void* d_temp_storage, |
| 931 | size_t& temp_storage_size_bytes, |
| 932 | auto keys_input, |
| 933 | auto values_input, |
| 934 | size_t run_length, |
| 935 | unsigned int* d_incorrect_flag, |
| 936 | size_t size, |
| 937 | hipStream_t stream, |
| 938 | bool debug_synchronous, |
| 939 | int /*seed_value*/) -> hipError_t |
| 940 | { |
| 941 | const check_run_inclusive_iterator output_it( |
| 942 | rocprim::make_tuple(run_length, d_incorrect_flag)); |
| 943 | |
| 944 | return rocprim::inclusive_scan_by_key(d_temp_storage, |
| 945 | temp_storage_size_bytes, |
| 946 | keys_input, |
| 947 | values_input, |
| 948 | output_it, |
| 949 | size, |
| 950 | rocprim::plus<size_t>{}, |
| 951 | rocprim::equal_to<size_t>{}, |
| 952 | stream, |
| 953 | debug_synchronous); |
| 954 | }; |
| 955 | large_indices_scan_by_key_test<decltype(inclusive_scan_by_key), UseGraphs>( |
| 956 | inclusive_scan_by_key); |
| 957 | } |
| 958 | |
| 959 | TEST(RocprimDeviceScanTests, LargeIndicesInclusiveScanByKey) |
| 960 | { |
no test coverage detected