Fixed bugs (#3)

ROCm · Nov 1, 2019 · 543437a · 543437a
1 parent 1137fe4
commit 543437a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 40 deletions.
diff --git a/src/imgaug/cpu/host_color_model_conversions.hpp b/src/imgaug/cpu/host_color_model_conversions.hpp
@@ -204,6 +204,11 @@ RppStatus vignette_host(T* srcPtr, RppiSize srcSize, T* dstPtr,
     }
 
     compute_multiply_host(srcPtr, maskFinal, srcSize, dstPtr, channel);
-
+
+    free(maskFinal);
+    free(kernelRows);
+    free(kernelColumns);
+    free(mask);
+
     return RPP_SUCCESS;
 }
diff --git a/src/imgaug/cpu/host_image_augmentations.hpp b/src/imgaug/cpu/host_image_augmentations.hpp
@@ -29,7 +29,8 @@ RppStatus blur_host(T* srcPtr, RppiSize srcSize, T* dstPtr,
     rppiKernelSize.height = kernelSize;
     rppiKernelSize.width = kernelSize;
     convolve_image_host(srcPtrMod, srcSizeMod, dstPtr, srcSize, kernel, rppiKernelSize, chnFormat, channel);
-
+    free(kernel);
+    free(srcPtrMod);
     return RPP_SUCCESS;
 }
 

diff --git a/src/include/cpu/rpp_cpu_common.hpp b/src/include/cpu/rpp_cpu_common.hpp
@@ -406,23 +406,6 @@ inline RppStatus generate_sobel_kernel_host(Rpp32f* kernel, Rpp32u type)
     return RPP_SUCCESS;
 }
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 // Kernels for functions
 
 template<typename T, typename U>
@@ -589,6 +572,7 @@ inline RppStatus resize_crop_kernel_host(T* srcPtr, RppiSize srcSize, T* dstPtr,
 
     resize_kernel_host(srcPtrResize, srcSizeSubImage, dstPtr, dstSize, chnFormat, channel);
 
+    free(srcPtrResize);
     return RPP_SUCCESS;
 
 }
@@ -727,7 +711,7 @@ inline RppStatus median_filter_kernel_host(T* srcPtrWindow, T* dstPtrPixel, Rppi
     std::sort(kernel, kernel + (kernelSize * kernelSize));
 
     *dstPtrPixel = *(kernel + (((kernelSize * kernelSize) - 1) / 2));
-
+    free(kernel);
     return RPP_SUCCESS;
 }
 
@@ -900,26 +884,6 @@ inline RppStatus non_max_suppression_kernel_host(T* srcPtrWindow, T* dstPtrPixel
 
     return RPP_SUCCESS;
 }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 // Convolution Functions
 
 template<typename T, typename U>