From a30bd536287d7853a98d690d954fff9674f410d1 Mon Sep 17 00:00:00 2001 From: Wayne Mitchell Date: Tue, 14 Feb 2023 08:55:37 -0800 Subject: [PATCH] Fix memory leak in sycl reduction boxloop (#836) Fix memory leak in sycl reduction boxloop --- src/struct_mv/_hypre_struct_mv.hpp | 2 ++ src/struct_mv/boxloop_sycl.h | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/struct_mv/_hypre_struct_mv.hpp b/src/struct_mv/_hypre_struct_mv.hpp index aa1b7c18e..839ae0095 100644 --- a/src/struct_mv/_hypre_struct_mv.hpp +++ b/src/struct_mv/_hypre_struct_mv.hpp @@ -1546,6 +1546,7 @@ else \ } \ }, hypre__tot, shared_sum_var); \ hypre_TMemcpy(&sum_var, shared_sum_var, HYPRE_Real, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE); \ + hypre_TFree(shared_sum_var, HYPRE_MEMORY_DEVICE); \ } /* Reduction BoxLoop2 */ @@ -1570,6 +1571,7 @@ else \ } \ }, hypre__tot, shared_sum_var); \ hypre_TMemcpy(&sum_var, shared_sum_var, HYPRE_Real, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE); \ + hypre_TFree(shared_sum_var, HYPRE_MEMORY_DEVICE); \ } /* Plain parallel_for loop */ diff --git a/src/struct_mv/boxloop_sycl.h b/src/struct_mv/boxloop_sycl.h index 8a68431cf..fded90e70 100644 --- a/src/struct_mv/boxloop_sycl.h +++ b/src/struct_mv/boxloop_sycl.h @@ -352,6 +352,7 @@ else \ } \ }, hypre__tot, shared_sum_var); \ hypre_TMemcpy(&sum_var, shared_sum_var, HYPRE_Real, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE); \ + hypre_TFree(shared_sum_var, HYPRE_MEMORY_DEVICE); \ } /* Reduction BoxLoop2 */ @@ -376,6 +377,7 @@ else \ } \ }, hypre__tot, shared_sum_var); \ hypre_TMemcpy(&sum_var, shared_sum_var, HYPRE_Real, 1, HYPRE_MEMORY_HOST, HYPRE_MEMORY_DEVICE); \ + hypre_TFree(shared_sum_var, HYPRE_MEMORY_DEVICE); \ } /* Plain parallel_for loop */