diff --git a/catch/hipTestMain/config/config_amd_linux b/catch/hipTestMain/config/config_amd_linux index 851180e680..612d118b5b 100644 --- a/catch/hipTestMain/config/config_amd_linux +++ b/catch/hipTestMain/config/config_amd_linux @@ -732,8 +732,6 @@ "=== SWDEV-511679 : Below tests fail in stress test ===", "Unit_hipIpcOpenMemHandle_Negative_Open_In_Two_Contexts_Same_Device", "Unit_hipIpcCloseMemHandle_Positive_Reference_Counting", - "=== SWDEV-508023 Below tests are temporarily disabled - Linux PSDB failed", - "Unit_hipDeviceGetGraphMemAttribute_Positive_DoubleMemory", "=== SWDEV-517063 Below tests are temporarily disabled due to PSDB failure", "Unit_hipGraphInstantiateWithFlags_FlagAutoFreeOnLaunch_check", "Unit_hipGraphInstantiateWithFlags_AutoFreeOnLaunchInLoop", diff --git a/catch/hipTestMain/config/config_amd_windows b/catch/hipTestMain/config/config_amd_windows index c5e96aafae..28298eb689 100644 --- a/catch/hipTestMain/config/config_amd_windows +++ b/catch/hipTestMain/config/config_amd_windows @@ -990,8 +990,6 @@ "Test Unit_hipGraphUserObj_ClonedGraph disabled due to SWDEV-483112", "Unit_hipGraphUserObj_ClonedGraph", "====================================================", - "=== SWDEV-508023 Below tests are temporarily disabled - windows PSDB failed", - "Unit_hipDeviceGetGraphMemAttribute_Positive_DoubleMemory", "=== SWDEV-517063 Below tests are temporarily disabled due to PSDB failure", "Unit_hipGraphInstantiateWithFlags_AutoFreeOnLaunchInLoop", "Unit_hipGraphInstantiateWithFlags_AutoFreeOnLaunchFillKernel", diff --git a/catch/unit/graph/hipDeviceGetGraphMemAttribute.cc b/catch/unit/graph/hipDeviceGetGraphMemAttribute.cc index 67d8b43b79..abf5600847 100644 --- a/catch/unit/graph/hipDeviceGetGraphMemAttribute.cc +++ b/catch/unit/graph/hipDeviceGetGraphMemAttribute.cc @@ -132,7 +132,10 @@ TEST_CASE("Unit_hipDeviceGetGraphMemAttribute_Positive_DoubleMemory") { HIP_CHECK(hipGraphExecDestroy(graph_exec1)); HIP_CHECK(hipGraphExecDestroy(graph_exec2)); + HIP_CHECK(hipDeviceGraphMemTrim(0)); + HIP_CHECK(hipStreamSynchronize(0)); + checkGraphMemAttribute(0, 2 * element_count * sizeof(int)); ResetGraphMemAttribute(); } @@ -170,7 +173,10 @@ TEST_CASE("Unit_hipDeviceGetGraphMemAttribute_Positive_ReuseMemory") { HIP_CHECK(hipGraphExecDestroy(graph_exec1)); HIP_CHECK(hipGraphExecDestroy(graph_exec2)); + HIP_CHECK(hipDeviceGraphMemTrim(0)); + HIP_CHECK(hipStreamSynchronize(0)); + checkGraphMemAttribute(0, element_count * sizeof(int)); ResetGraphMemAttribute(); #endif