[rocPRIM] Fix memory leaks in different unit tests

[NguyenNhuDi]

Motivation

TCT reported multiple memory leaks coming from different rocPRIM unit tests. Specifically:

• test_device_merge_inplace.cpp
• test_device_transform.cpp
• test_hipgraph_basic.cpp

Technical Details

In test_device_merge_inplace.cpp in the Basic test, d_temp_storage was never freed manually. We added a call to free_manually() to address this issue.

In test_device_transform.cpp in the UnalignedPointer test, d_unaligned and d_input were allocated but never freed. We added a call to hipFree at the end of the test to address this issue.

In test_hipgraph_basic.cpp the wrong call to free for d_data was used. Instead of calling hipFree, hipFreeAsync should be called instead since it was allocated with hipMallocAsync. Also the order of freeing was not correct, and needed to be rearranged.

Test Plan

Run valgrind with these tests locally to check for test failures and memory leaks. As well as CI checks.

Test Result

When valgrind was ran locally no failures and leaks are observed.

[umfranzw]

This looks good to me, pending CI validation.

[NguyenNhuDi]

Hi @umfranzw and @amd-hsong I had to make a hotfix since an unit test was failing due to my changes. If I could get your approvals again that would be great