AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-401781 - Enhance test case.

Browse Source 
Change-Id: I47074cc972512ac1b699ed4fc9386d470fae1c61
This commit is contained in:
Jaydeep Patel
2023-12-18 10:02:56 +00:00
committed by Rakesh Roy
parent aefccff870
commit 8993dd87ef
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
catch/unit/graph/hipGraphInstantiateWithFlags.cc
+26
View File
@@ -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));