diff --git a/projects/rocprim/test/hipgraph/test_hipgraph_basic.cpp b/projects/rocprim/test/hipgraph/test_hipgraph_basic.cpp index cc473d680e06..13849d712e8a 100644 --- a/projects/rocprim/test/hipgraph/test_hipgraph_basic.cpp +++ b/projects/rocprim/test/hipgraph/test_hipgraph_basic.cpp @@ -99,10 +99,10 @@ void testStreamCapture() ASSERT_EQ(h_data, num_launches); // Clean up + HIP_CHECK(hipFreeAsync(d_data, stream)); + HIP_CHECK(hipStreamDestroy(stream)); HIP_CHECK(hipGraphDestroy(graph)); HIP_CHECK(hipGraphExecDestroy(instance)); - HIP_CHECK(hipFree(d_data)); - HIP_CHECK(hipStreamDestroy(stream)); } void testManualConstruction() @@ -159,10 +159,10 @@ void testManualConstruction() ASSERT_EQ(h_data, num_launches); // Clean up + HIP_CHECK(hipFreeAsync(d_data, stream)); + HIP_CHECK(hipStreamDestroy(stream)); HIP_CHECK(hipGraphDestroy(graph)); HIP_CHECK(hipGraphExecDestroy(instance)); - HIP_CHECK(hipFree(d_data)); - HIP_CHECK(hipStreamDestroy(stream)); } void testStreamCaptureWithAtomics() @@ -217,10 +217,10 @@ void testStreamCaptureWithAtomics() ASSERT_EQ(h_data, num_launches * num_blocks * num_threads); // Clean up + HIP_CHECK(hipFreeAsync(d_data, stream)); + HIP_CHECK(hipStreamDestroy(stream)); HIP_CHECK(hipGraphDestroy(graph)); HIP_CHECK(hipGraphExecDestroy(instance)); - HIP_CHECK(hipFree(d_data)); - HIP_CHECK(hipStreamDestroy(stream)); } TEST(TestHipGraphBasic, CaptureFromStream) diff --git a/projects/rocprim/test/rocprim/test_device_merge_inplace.cpp b/projects/rocprim/test/rocprim/test_device_merge_inplace.cpp index d6fd85d61e70..0d648fa289ca 100644 --- a/projects/rocprim/test/rocprim/test_device_merge_inplace.cpp +++ b/projects/rocprim/test/rocprim/test_device_merge_inplace.cpp @@ -90,6 +90,7 @@ TEST(RocprimDeviceMergeInplaceTests, Basic) h_data = d_data.load(); d_data.free_manually(); + d_temp_storage.free_manually(); ASSERT_NO_FATAL_FAILURE(test_utils::assert_eq(h_data, h_expected)); } diff --git a/projects/rocprim/test/rocprim/test_device_transform.cpp b/projects/rocprim/test/rocprim/test_device_transform.cpp index 514b465720b9..96eb9d829112 100644 --- a/projects/rocprim/test/rocprim/test_device_transform.cpp +++ b/projects/rocprim/test/rocprim/test_device_transform.cpp @@ -596,6 +596,8 @@ TEST(RocprimDeviceTransformTests, UnalignedPointer) // Check if output values are as expected ASSERT_NO_FATAL_FAILURE( test_utils::assert_near(output, expected, test_utils::precision)); + + HIP_CHECK(hipFree(d_unaligned)); } } }