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

oneapi-src · Nov 7, 2024 · a64c93f · a64c93f
1 parent f49cc0b
commit a64c93f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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(