diff --git a/test/conformance/device/urDeviceGet.cpp b/test/conformance/device/urDeviceGet.cpp index 9b053f1f46..85a4818d09 100644 --- a/test/conformance/device/urDeviceGet.cpp +++ b/test/conformance/device/urDeviceGet.cpp @@ -56,6 +56,6 @@ TEST_F(urDeviceGetTest, InvalidValueNumEntries) { ASSERT_NE(count, 0); std::vector devices(count); ASSERT_EQ_RESULT( - UR_RESULT_ERROR_INVALID_VALUE, + UR_RESULT_ERROR_INVALID_SIZE, urDeviceGet(platform, UR_DEVICE_TYPE_ALL, 0, devices.data(), nullptr)); } diff --git a/test/conformance/device/urDeviceGetInfo.cpp b/test/conformance/device/urDeviceGetInfo.cpp index bbe08fe6d8..eb8321b7b4 100644 --- a/test/conformance/device/urDeviceGetInfo.cpp +++ b/test/conformance/device/urDeviceGetInfo.cpp @@ -13,7 +13,6 @@ static std::unordered_map device_info_size_map = { {UR_DEVICE_INFO_MAX_COMPUTE_UNITS, sizeof(uint32_t)}, {UR_DEVICE_INFO_MAX_WORK_ITEM_DIMENSIONS, sizeof(uint32_t)}, {UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE, sizeof(size_t)}, - {UR_DEVICE_INFO_MAX_WORK_GROUP_SIZE, sizeof(size_t)}, {UR_DEVICE_INFO_SINGLE_FP_CONFIG, sizeof(ur_device_fp_capability_flags_t)}, {UR_DEVICE_INFO_HALF_FP_CONFIG, sizeof(ur_device_fp_capability_flags_t)}, {UR_DEVICE_INFO_DOUBLE_FP_CONFIG, sizeof(ur_device_fp_capability_flags_t)}, @@ -258,7 +257,7 @@ TEST_P(urDeviceGetInfoTest, Success) { ASSERT_SUCCESS(urDeviceGetInfo(device, info_type, size, info_data.data(), nullptr)); } else { - ASSERT_EQ_RESULT(result, UR_RESULT_ERROR_INVALID_ENUMERATION); + ASSERT_EQ_RESULT(result, UR_RESULT_ERROR_UNSUPPORTED_ENUMERATION); } } } diff --git a/test/conformance/device/urDeviceRelease.cpp b/test/conformance/device/urDeviceRelease.cpp index 300f646851..3b5e74a575 100644 --- a/test/conformance/device/urDeviceRelease.cpp +++ b/test/conformance/device/urDeviceRelease.cpp @@ -26,6 +26,12 @@ TEST_F(urDeviceReleaseTest, Success) { TEST_F(urDeviceReleaseTest, SuccessSubdevices) { for (auto device : devices) { + if (!uur::hasDevicePartitionSupport(device, + UR_DEVICE_PARTITION_EQUALLY)) { + ::testing::Message() << "Device: \'" << device + << "\' does not support partitioning equally."; + continue; + } ur_device_partition_property_t prop = uur::makePartitionEquallyDesc(1); diff --git a/test/conformance/device/urDeviceRetain.cpp b/test/conformance/device/urDeviceRetain.cpp index e5999ba24b..77a9c58b79 100644 --- a/test/conformance/device/urDeviceRetain.cpp +++ b/test/conformance/device/urDeviceRetain.cpp @@ -27,6 +27,12 @@ TEST_F(urDeviceRetainTest, Success) { TEST_F(urDeviceRetainTest, SuccessSubdevices) { for (auto device : devices) { + if (!uur::hasDevicePartitionSupport(device, + UR_DEVICE_PARTITION_EQUALLY)) { + ::testing::Message() << "Device: \'" << device + << "\' does not support partitioning equally."; + continue; + } ur_device_partition_property_t prop = uur::makePartitionEquallyDesc(1); ur_device_partition_properties_t properties{ diff --git a/test/conformance/device/urDeviceSelectBinary.cpp b/test/conformance/device/urDeviceSelectBinary.cpp index 34d008a3cd..44d32e3c6d 100644 --- a/test/conformance/device/urDeviceSelectBinary.cpp +++ b/test/conformance/device/urDeviceSelectBinary.cpp @@ -56,7 +56,7 @@ TEST_F(urDeviceSelectBinaryTest, InvalidValueNumBinaries) { for (auto device : devices) { uint32_t selected_binary; ASSERT_EQ_RESULT( - UR_RESULT_ERROR_INVALID_VALUE, + UR_RESULT_ERROR_INVALID_SIZE, urDeviceSelectBinary(device, binaries, 0, &selected_binary)); } }