COMP: Add missing iostream include for CUDA classes

SimonRit · Mar 12, 2021 · 3dba988 · 3dba988
1 parent 4b3d3e9
commit 3dba988
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rtkCudaLagCorrectionImageFilter.cu b/src/rtkCudaLagCorrectionImageFilter.cu
@@ -114,7 +114,7 @@ CUDA_lag_correction(int              proj_idx_in[3],      // overlapping input r
   cudaMalloc((void **)&d_state, state_size);
   if (cudaMemcpy(d_state, h_state, state_size, cudaMemcpyHostToDevice) != cudaSuccess)
   {
-    std::cout << "Error allocating state" << std::endl;
+    itkGenericExceptionMacro("Error allocating state");
   }
 
   if (coefficients[8] <= 0.0)

diff --git a/src/rtkCudaPolynomialGainCorrectionImageFilter.cu b/src/rtkCudaPolynomialGainCorrectionImageFilter.cu
@@ -111,7 +111,7 @@ CUDA_gain_correction(int              proj_idx_in[3],      // overlapping input
   cudaMalloc((void **)&d_powerlut, lut_size);
   if (cudaMemcpy(d_powerlut, h_powerlut, lut_size, cudaMemcpyHostToDevice) != cudaSuccess)
   {
-    std::cout << "Error allocating state" << std::endl;
+    itkGenericExceptionMacro("Error allocating state");
   }
 
   cudaMemcpyToSymbol(cst_coef, coefficients, 2 * sizeof(float));