SWDEV-411064 - fix device mem size to accurately match memory size from same device

Change-Id: I9caee84c58312053a7b387c8516b386d53c2197f


[ROCm/hip-tests commit: c340f47ac1]

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);