From 4d588564ad6f37bde7f7e951d0a6497fd73713d3 Mon Sep 17 00:00:00 2001
From: Maneesh Gupta <maneesh.gupta@amd.com>
Date: Mon, 10 Oct 2022 16:15:08 +0530
Subject: [PATCH] SWDEV-1 - Add missing checks in memset.cc

---
 catch/unit/memory/memset.cc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/catch/unit/memory/memset.cc b/catch/unit/memory/memset.cc
index 677d8a2d85..91f2d04601 100644
--- a/catch/unit/memory/memset.cc
+++ b/catch/unit/memory/memset.cc
@@ -6,7 +6,8 @@ TEST_CASE("Unit_hipMemset_4bytes") {
   REQUIRE(res == hipSuccess);
   res = hipMemset(d_a, 0, sizeof(int));
   REQUIRE(res == hipSuccess);
-  hipFree(d_a);
+  res = hipFree(d_a);
+  REQUIRE(res == hipSuccess);
 }
 
 TEST_CASE("Unit_hipMemset_4bytes_hostMem") {
@@ -15,5 +16,6 @@ TEST_CASE("Unit_hipMemset_4bytes_hostMem") {
   REQUIRE(res == hipSuccess);
   res = hipMemset(d_a, 0, sizeof(int));
   REQUIRE(res == hipSuccess);
-  hipHostFree(d_a);
+  res = hipHostFree(d_a);
+  REQUIRE(res == hipSuccess);
 }