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

Change-Id: I9caee84c58312053a7b387c8516b386d53c2197f [ROCm/hip-tests commit: c340f47ac1]
2023-07-27 14:19:31 +01:00
parent cb193f70c4
commit 668bcece7a
1 changed files with 1 additions and 0 deletions
@@ -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);