Skip to content

Commit 5ce5084

Browse files
authored
SWDEV-546111 - Fix memory leaks in coopGroup tests (#509)
1 parent 3125b51 commit 5ce5084

File tree

3 files changed

+12
-0
lines changed

3 files changed

+12
-0
lines changed

catch/unit/cooperativeGrps/coalesced_groups_shfl_down_old.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,7 @@ static void test_shfl_down() {
233233

234234
HIPCHECK(hipHostFree(hPtr));
235235
HIPCHECK(hipFree(dPtr));
236+
HIPCHECK(hipFree(dResults));
236237
free(cpuResultsArr);
237238
}
238239
}

catch/unit/cooperativeGrps/coalesced_groups_shfl_up_old.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,7 @@ static void test_shfl_up() {
221221

222222
HIPCHECK(hipHostFree(hPtr));
223223
HIPCHECK(hipFree(dPtr));
224+
HIPCHECK(hipFree(dResults));
224225
free(cpuResultsArr);
225226
}
226227
}

catch/unit/cooperativeGrps/hipCGCoalescedGroups_old.cc

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -394,6 +394,8 @@ static void test_shfl_any_to_any() {
394394

395395
HIPCHECK(hipHostFree(hPtr));
396396
HIPCHECK(hipFree(dPtr));
397+
HIPCHECK(hipFree(dResults));
398+
HIPCHECK(hipFree(dsrcArr));
397399
free(srcArr);
398400
free(srcArrCpu);
399401
free(cpuResultsArr);
@@ -461,6 +463,7 @@ static void test_shfl_broadcast() {
461463

462464
HIPCHECK(hipHostFree(hPtr));
463465
HIPCHECK(hipFree(dPtr));
466+
HIPCHECK(hipFree(dResults));
464467
free(cpuResultsArr);
465468
}
466469
}
@@ -554,4 +557,11 @@ TEST_CASE("Unit_coalesced_groups") {
554557

555558
std::cout << "Now grouping active threads based on branch divergence" << '\n' << std::endl;
556559
test_active_threads_grouping();
560+
561+
HIPCHECK(hipFree(d_data_to_filter));
562+
HIPCHECK(hipFree(d_filtered_data));
563+
HIPCHECK(hipFree(d_nres));
564+
free(data_to_filter);
565+
free(filtered_data);
566+
free(host_filtered_data);
557567
}

0 commit comments

Comments
 (0)