cuda reduce kernel

LLNL · May 5, 2023 · b94b91e · b94b91e
1 parent ef33bcc
commit b94b91e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/LinAlg/VectorCudaKernels.cu b/src/LinAlg/VectorCudaKernels.cu
@@ -393,9 +393,9 @@ __global__ void add_linear_damping_term_cu(int n, double* data, const double* ix
 }
 
 /** @brief y[i] = 1.0 if x[i] is positive and id[i] = 1.0, otherwise y[i] = 0 */
-__global__ void is_posive_w_pattern_cu(int n, double* data, const double* vd, const double* id)
+__global__ void is_posive_w_pattern_cu(int n, int* data, const double* vd, const double* id)
 {
-  extern __shared__ float shared_sum[];
+  extern __shared__ int shared_sum[];
   const int num_threads = blockDim.x * gridDim.x;
   const int tid = blockIdx.x * blockDim.x + threadIdx.x;
   int sum = 0;
@@ -957,7 +957,7 @@ int is_posive_w_pattern_kernel(int n_local,
   cudaMemcpy(h_retval, d_retval, num_blocks*sizeof(int), cudaMemcpyDeviceToHost);
 
   int sum_result = 0;
-  for(int i=0;i<block_size;i++) {
+  for(int i=0;i<num_blocks;i++) {
     sum_result += h_retval[i];
   }
 
@@ -1242,9 +1242,10 @@ int all_positive_w_pattern_kernel(int n, const double* d1, const double* id)
   // TODO: how to avoid this temp vec?
 //  thrust::device_vector<double> v_temp(n);
 //  double* dv_ptr = thrust::raw_pointer_cast(v_temp.data());
+//  is_posive_w_pattern_kernel(n, dv_ptr, d1, id);
 //  return thrust::reduce(thrust::device, v_temp.begin(), v_temp.end(), (int)0, thrust::plus<int>());
 
-  int irev = hiop::cuda::is_posive_w_pattern_kernel(n, dv_ptr, d1, id);
+  int irev = hiop::cuda::is_posive_w_pattern_kernel(n, d1, id);
   return irev;
 }