@@ -61,11 +61,11 @@ void runBenchmark(long max_work) {
6161 // Request and allocate temporary storage
6262 void *d_temp_storage = nullptr ;
6363 size_t temp_storage_bytes = 0 ;
64- hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 );
64+ HIP_CHECK ( hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 ) );
6565 HIP_CHECK (hipMalloc ((void **)&d_temp_storage, temp_storage_bytes));
6666
6767 // Run
68- hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 );
68+ HIP_CHECK ( hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 ) );
6969
7070 // copy results back to the host
7171 unsigned long long int h_count = 0 ;
@@ -127,11 +127,11 @@ void runBenchmarkTime(long max_work, int runtime_in_seconds) {
127127 // Request and allocate temporary storage
128128 void *d_temp_storage = nullptr ;
129129 size_t temp_storage_bytes = 0 ;
130- hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 );
130+ HIP_CHECK ( hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 ) );
131131 HIP_CHECK (hipMalloc ((void **)&d_temp_storage, temp_storage_bytes));
132132
133133 // Run
134- hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 );
134+ HIP_CHECK ( hipcub::DeviceReduce::Sum (d_temp_storage, temp_storage_bytes, d_count, d_out, 256 ) );
135135
136136 // copy results back to the host
137137 unsigned long long int h_count = 0 ;
0 commit comments