From 8993dd87efdf805570d0039f193eaeafb560a6e5 Mon Sep 17 00:00:00 2001 From: Jaydeep Patel Date: Mon, 18 Dec 2023 10:02:56 +0000 Subject: [PATCH] SWDEV-401781 - Enhance test case. Change-Id: I47074cc972512ac1b699ed4fc9386d470fae1c61 --- .../graph/hipGraphInstantiateWithFlags.cc | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/catch/unit/graph/hipGraphInstantiateWithFlags.cc b/catch/unit/graph/hipGraphInstantiateWithFlags.cc index 54210442fb..9571962efe 100644 --- a/catch/unit/graph/hipGraphInstantiateWithFlags.cc +++ b/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));