[SYCL][DeviceSanitizer] Fix memory leaks in e2e tests (#13540)

Author: AllanZyne

sycl/test-e2e/AddressSanitizer/common/kernel-debug.cpp

@@ -15,6 +15,7 @@ int main() {
   // CHECK-DEBUG: [kernel]
   // CHECK-NOT: [kernel]
 
+  sycl::free(array, Q);
   std::cout << "PASS" << std::endl;
   return 0;
 }

sycl/test-e2e/AddressSanitizer/out-of-bounds/USM/parallel_for_char.cpp

@@ -21,12 +21,8 @@ int main() {
   auto *array = sycl::malloc_host<char>(N, Q);
 #elif defined(MALLOC_SHARED)
   auto *array = sycl::malloc_shared<char>(N, Q);
-#elif defined(MALLOC_DEVICE)
+#else // defined(MALLOC_DEVICE)
   auto *array = sycl::malloc_device<char>(N, Q);
-#elif defined(MALLOC_SYSTEM)
-  auto *array = new char[N];
-#else
-#error "Must provide malloc type to run the test"
 #endif
 
   Q.submit([&](sycl::handler &h) {
@@ -41,5 +37,6 @@ int main() {
   // CHECK: {{READ of size 1 at kernel <.*MyKernelR_4> LID\(0, 0, 0\) GID\(12345, 0, 0\)}}
   // CHECK: {{  #0 .* .*parallel_for_char.cpp:}}[[@LINE-7]]
 
+  sycl::free(array, Q);
   return 0;
 }

sycl/test-e2e/AddressSanitizer/out-of-bounds/USM/parallel_for_double.cpp

@@ -21,12 +21,8 @@ int main() {
   auto *array = sycl::malloc_host<double>(N, Q);
 #elif defined(MALLOC_SHARED)
   auto *array = sycl::malloc_shared<double>(N, Q);
-#elif defined(MALLOC_DEVICE)
+#else // defined(MALLOC_DEVICE)
   auto *array = sycl::malloc_device<double>(N, Q);
-#elif defined(MALLOC_SYSTEM)
-  auto *array = new double[N];
-#else
-#error "Must provide malloc type to run the test"
 #endif
 
   Q.submit([&](sycl::handler &h) {
@@ -41,5 +37,6 @@ int main() {
   // CHECK: {{READ of size 8 at kernel <.*MyKernelR_4> LID\(0, 0, 0\) GID\(123456, 0, 0\)}}
   // CHECK: {{  #0 .* .*parallel_for_double.cpp:}}[[@LINE-7]]
 
+  sycl::free(array, Q);
   return 0;
 }

sycl/test-e2e/AddressSanitizer/out-of-bounds/USM/parallel_for_func.cpp

@@ -28,12 +28,8 @@ int main() {
   auto *array = sycl::malloc_host<int>(N, Q);
 #elif defined(MALLOC_SHARED)
   auto *array = sycl::malloc_shared<int>(N, Q);
-#elif defined(MALLOC_DEVICE)
+#else // defined(MALLOC_DEVICE)
   auto *array = sycl::malloc_device<int>(N, Q);
-#elif defined(MALLOC_SYSTEM)
-  auto *array = new int[N];
-#else
-#error "Must provide malloc type to run the test"
 #endif
 
   Q.submit([&](sycl::handler &h) {
@@ -43,5 +39,6 @@ int main() {
   });
   Q.wait();
 
+  sycl::free(array, Q);
   return 0;
 }

sycl/test-e2e/AddressSanitizer/out-of-bounds/USM/parallel_for_int.cpp

@@ -21,12 +21,8 @@ int main() {
   auto *array = sycl::malloc_host<int>(N, Q);
 #elif defined(MALLOC_SHARED)
   auto *array = sycl::malloc_shared<int>(N, Q);
-#elif defined(MALLOC_DEVICE)
+#else // defined(MALLOC_DEVICE)
   auto *array = sycl::malloc_device<int>(N, Q);
-#elif defined(MALLOC_SYSTEM)
-  auto *array = new int[N];
-#else
-#error "Must provide malloc type to run the test"
 #endif
 
   Q.submit([&](sycl::handler &h) {
@@ -41,5 +37,6 @@ int main() {
   // CHECK: {{READ of size 4 at kernel <.*MyKernelR_4> LID\(0, 0, 0\) GID\(1234567, 0, 0\)}}
   // CHECK: {{  #0 .* .*parallel_for_int.cpp:}}[[@LINE-7]]
 
+  sycl::free(array, Q);
   return 0;
 }

sycl/test-e2e/AddressSanitizer/out-of-bounds/USM/parallel_for_short.cpp

@@ -21,12 +21,8 @@ int main() {
   auto *array = sycl::malloc_host<short>(N, Q);
 #elif defined(MALLOC_SHARED)
   auto *array = sycl::malloc_shared<short>(N, Q);
-#elif defined(MALLOC_DEVICE)
+#else // defined(MALLOC_DEVICE)
   auto *array = sycl::malloc_device<short>(N, Q);
-#elif defined(MALLOC_SYSTEM)
-  auto *array = new short[N];
-#else
-#error "Must provide malloc type to run the test"
 #endif
 
   Q.submit([&](sycl::handler &h) {
@@ -41,5 +37,6 @@ int main() {
   // CHECK: {{READ of size 2 at kernel <.*MyKernelR_4> LID\(0, 0, 0\) GID\(123456789, 0, 0\)}}
   // CHECK: {{  #0 .* .*parallel_for_short.cpp:}}[[@LINE-7]]
 
+  sycl::free(array, Q);
   return 0;
 }