From 668bcece7acf1f0cd2a2babfe5bff6c8e68dabe0 Mon Sep 17 00:00:00 2001
From: Jatin Chaudhary <JatinJaikishan.Chaudhary@amd.com>
Date: Thu, 27 Jul 2023 14:19:31 +0100
Subject: [PATCH] SWDEV-411064 - fix device mem size to accurately match memory
 size from same device

Change-Id: I9caee84c58312053a7b387c8516b386d53c2197f


[ROCm/hip-tests commit: c340f47ac192069b2b675b24e71c5e2548cab41e]
---
 projects/hip-tests/catch/unit/device/hipDeviceTotalMem.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/hip-tests/catch/unit/device/hipDeviceTotalMem.cc b/projects/hip-tests/catch/unit/device/hipDeviceTotalMem.cc
index 7c21019424..fa2014bfe0 100644
--- a/projects/hip-tests/catch/unit/device/hipDeviceTotalMem.cc
+++ b/projects/hip-tests/catch/unit/device/hipDeviceTotalMem.cc
@@ -95,6 +95,7 @@ TEST_CASE("Unit_hipDeviceTotalMem_ValidateTotalMem") {
   HIP_CHECK(hipDeviceTotalMem(&totMem, device));
 
   size_t free = 0, total = 0;
+  HIP_CHECK(hipSetDevice(devNo));
   HIP_CHECK(hipMemGetInfo(&free, &total));
 
   REQUIRE(totMem == prop.totalGlobalMem);