Check for unsupported on CreateWithNative as well as GetNative.

Author: aarongreig

test/conformance/memory/urMemBufferCreateWithNativeHandle.cpp

@@ -19,7 +19,7 @@ TEST_P(urMemBufferCreateWithNativeHandleTest, Success) {
     // We can however convert the native_handle back into a unified-runtime handle
     // and perform some query on it to verify that it works.
     ur_mem_handle_t mem = nullptr;
-    ASSERT_SUCCESS(
+    UUR_ASSERT_SUCCESS_OR_UNSUPPORTED(
         urMemBufferCreateWithNativeHandle(hNativeMem, context, nullptr, &mem));
     ASSERT_NE(mem, nullptr);
 
@@ -44,7 +44,7 @@ TEST_P(urMemBufferCreateWithNativeHandleTest, SuccessWithProperties) {
         /*.pNext =*/nullptr,
         /*.isNativeHandleOwned =*/false,
     };
-    ASSERT_SUCCESS(
+    UUR_ASSERT_SUCCESS_OR_UNSUPPORTED(
         urMemBufferCreateWithNativeHandle(hNativeMem, context, &props, &mem));
     ASSERT_NE(mem, nullptr);
 

test/conformance/memory/urMemImageCreateWithNativeHandle.cpp

@@ -14,7 +14,7 @@ TEST_P(urMemImageCreateWithNativeHandleTest, Success) {
         urMemGetNativeHandle(image, device, &native_handle));
 
     ur_mem_handle_t mem = nullptr;
-    ASSERT_SUCCESS(urMemImageCreateWithNativeHandle(
+    UUR_ASSERT_SUCCESS_OR_UNSUPPORTED(urMemImageCreateWithNativeHandle(
         native_handle, context, &image_format, &image_desc, nullptr, &mem));
     ASSERT_NE(nullptr, mem);
 
@@ -36,7 +36,7 @@ TEST_P(urMemImageCreateWithNativeHandleTest, SuccessWithProperties) {
     // We can't pass isNativeHandleOwned = true in the generic tests since
     // we always get the native handle from a UR object, and transferring
     // ownership from one UR object to another isn't allowed.
-    ASSERT_SUCCESS(urMemImageCreateWithNativeHandle(
+    UUR_ASSERT_SUCCESS_OR_UNSUPPORTED(urMemImageCreateWithNativeHandle(
         native_handle, context, &image_format, &image_desc, &props, &mem));
     ASSERT_NE(nullptr, mem);
 

test/conformance/testing/include/uur/fixtures.h

@@ -23,11 +23,13 @@
     (void)0
 
 #define UUR_ASSERT_SUCCESS_OR_UNSUPPORTED(ret)                                 \
-    auto status = ret;                                                         \
-    if (status == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {                       \
-        GTEST_SKIP();                                                          \
-    } else {                                                                   \
-        ASSERT_EQ(status, UR_RESULT_SUCCESS);                                  \
+    {                                                                          \
+        auto status = ret;                                                     \
+        if (status == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {                   \
+            GTEST_SKIP();                                                      \
+        } else {                                                               \
+            ASSERT_EQ(status, UR_RESULT_SUCCESS);                              \
+        }                                                                      \
     }
 
 namespace uur {