Change initial GetInfo assert to get ref count is >= 1.

martygrant · martygrant · commit 5ba5c2fc94e0 · 2024-11-14T15:01:30.000Z
diff --git a/test/conformance/virtual_memory/urPhysicalMemRelease.cpp b/test/conformance/virtual_memory/urPhysicalMemRelease.cpp
@@ -12,7 +12,7 @@ TEST_P(urPhysicalMemReleaseTest, Success) {
     ASSERT_SUCCESS(
         urPhysicalMemGetInfo(physical_mem, UR_PHYSICAL_MEM_INFO_REFERENCE_COUNT,
                              sizeof(referenceCount), &referenceCount, nullptr));
-    ASSERT_EQ(referenceCount, 1);
+    ASSERT_GE(1, referenceCount);
 
     ASSERT_SUCCESS(urPhysicalMemRetain(physical_mem));
     ASSERT_SUCCESS(
diff --git a/test/conformance/virtual_memory/urPhysicalMemRetain.cpp b/test/conformance/virtual_memory/urPhysicalMemRetain.cpp
@@ -12,7 +12,7 @@ TEST_P(urPhysicalMemRetainTest, Success) {
     ASSERT_SUCCESS(
         urPhysicalMemGetInfo(physical_mem, UR_PHYSICAL_MEM_INFO_REFERENCE_COUNT,
                              sizeof(referenceCount), &referenceCount, nullptr));
-    ASSERT_EQ(referenceCount, 1);
+    ASSERT_GE(1, referenceCount);
 
     ASSERT_SUCCESS(urPhysicalMemRetain(physical_mem));
     ASSERT_SUCCESS(
