SWDEV-401781 - Enhance test case.

Change-Id: I47074cc972512ac1b699ed4fc9386d470fae1c61


[ROCm/hip-tests commit: 8993dd87ef]

projects/hip-tests/catch/unit/graph/hipGraphInstantiateWithFlags.cc

@@ -348,6 +348,32 @@ TEST_CASE("Unit_hipGraphInstantiateWithFlags_FlagAutoFreeOnLaunch_check") {
   HIP_CHECK(hipGraphLaunch(graphExec, stream));
   HIP_CHECK(hipStreamSynchronize(stream));
 
+  size_t bmem = 0, bmemres = 0;
+  HIP_CHECK(hipDeviceGraphMemTrim(0));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrUsedMemCurrent, &bmem));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrReservedMemCurrent, &bmemres));
+
+  HIP_CHECK(hipGraphLaunch(graphExec, stream));
+  HIP_CHECK(hipStreamSynchronize(stream));
+
+  size_t amem = 0, amemres = 0;
+  HIP_CHECK(hipDeviceGraphMemTrim(0));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrUsedMemCurrent, &amem));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrReservedMemCurrent, &amemres));
+
+  REQUIRE(bmem == amem);
+  REQUIRE(bmemres == amemres);
+
+  HIP_CHECK(hipGraphLaunch(graphExec, stream));
+  HIP_CHECK(hipStreamSynchronize(stream));
+
+  HIP_CHECK(hipDeviceGraphMemTrim(0));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrUsedMemCurrent, &amem));
+  HIP_CHECK(hipDeviceGetGraphMemAttribute(0, hipGraphMemAttrReservedMemCurrent, &amemres));
+
+  REQUIRE(bmem == amem);
+  REQUIRE(bmemres == amemres);
+
   HIP_CHECK(hipFree(A_d));  //  free allocMemory manually
   HIP_CHECK(hipGraphDestroy(graph));
   HIP_CHECK(hipGraphExecDestroy(graphExec));