Merge pull request #1159 from beomki-yeo/fix-array-wrapper-test

beomki-yeo · web-flow · commit ed92914693d4 · 2025-09-24T10:07:09.000+02:00
Free the memory properly in `array-wrapper-test`
diff --git a/tests/cuda/test_array_wrapper.cu b/tests/cuda/test_array_wrapper.cu
@@ -91,6 +91,8 @@ TEST(CUDAArrayWrapper, SoALayout) {
               cudaSuccess);
 
     ASSERT_EQ(host_result, n);
+
+    ASSERT_EQ(cudaFree(dev_result), cudaSuccess);
 }
 
 TEST(CUDAArrayWrapper, AoSLayout) {
@@ -124,4 +126,6 @@ TEST(CUDAArrayWrapper, AoSLayout) {
               cudaSuccess);
 
     ASSERT_EQ(host_result, n);
+
+    ASSERT_EQ(cudaFree(dev_result), cudaSuccess);
 }

Original file line number	Diff line number	Diff line change
`@@ -91,6 +91,8 @@ TEST(CUDAArrayWrapper, SoALayout) {`
`91`	`91`	`cudaSuccess);`
`92`	`92`
`93`	`93`	`ASSERT_EQ(host_result, n);`
	`94`	`+`
	`95`	`+ ASSERT_EQ(cudaFree(dev_result), cudaSuccess);`
`94`	`96`	`}`
`95`	`97`
`96`	`98`	`TEST(CUDAArrayWrapper, AoSLayout) {`
`@@ -124,4 +126,6 @@ TEST(CUDAArrayWrapper, AoSLayout) {`
`124`	`126`	`cudaSuccess);`
`125`	`127`
`126`	`128`	`ASSERT_EQ(host_result, n);`
	`129`	`+`
	`130`	`+ ASSERT_EQ(cudaFree(dev_result), cudaSuccess);`
`127`	`131`	`}`