Media
/
compute-runtime
mirror of https://github.com/intel/compute-runtime.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feature: add ze_image_query_allocation support 

Related-To: NEO-9988

Signed-off-by: Jakub Nowacki <jakub.nowacki@intel.com>
This commit is contained in:
Jakub Nowacki 2025-05-06 13:03:56 +00:00 committed by Compute-Runtime-Automation
parent 6afefa1e31
commit 38b178dd9e
2 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
level_zero/core/source/driver/driver_handle_imp_helper.cpp
View File

@ -47,6 +47,7 @@ const std::vector<std::pair<std::string, uint32_t>> DriverHandleImp::extensionsS
{ZE_INTEL_DEVICE_BLOCK_ARRAY_EXP_NAME, ZE_INTEL_DEVICE_BLOCK_ARRAY_EXP_PROPERTIES_VERSION_CURRENT},
{ZE_INTEL_KERNEL_GET_PROGRAM_BINARY_EXP_NAME, ZE_INTEL_KERNEL_GET_PROGRAM_BINARY_EXP_VERSION_CURRENT},
{ZE_IMAGE_MEMORY_PROPERTIES_EXP_NAME, ZE_IMAGE_MEMORY_PROPERTIES_EXP_VERSION_CURRENT},
{ZE_IMAGE_QUERY_ALLOC_PROPERTIES_EXT_NAME, ZE_IMAGE_QUERY_ALLOC_PROPERTIES_EXT_VERSION_CURRENT},
// Metrics Driver experimental extensions
{ZET_INTEL_METRIC_APPEND_MARKER_EXP_NAME, ZET_INTEL_METRIC_APPEND_MARKER_EXP_VERSION_CURRENT},

18
level_zero/core/test/unit_tests/sources/driver/test_driver.cpp
View File

@ -1687,5 +1687,23 @@ TEST_F(DriverExtensionsTest, givenSupportedExtensionsWhenCheckIfImageMemoryPrope
EXPECT_EQ(expectedVersion, (*it).version);
}
TEST_F(DriverExtensionsTest, givenSupportedExtensionsWhenCheckIfImageQueryAllocationIsSupportedThenCorrectResultsAreReturned) {
uint32_t count = 0;
ze_result_t res = driverHandle->getExtensionProperties(&count, nullptr);
EXPECT_NE(0u, count);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
std::vector<ze_driver_extension_properties_t> extensionProperties;
extensionProperties.resize(count);
res = driverHandle->getExtensionProperties(&count, extensionProperties.data());
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
auto it = std::find_if(extensionProperties.begin(), extensionProperties.end(), [](const auto &extension) { return (strcmp(extension.name, ZE_IMAGE_QUERY_ALLOC_PROPERTIES_EXT_NAME) == 0); });
EXPECT_NE(it, extensionProperties.end());
uint32_t expectedVersion = ZE_MAKE_VERSION(1, 0);
EXPECT_EQ(expectedVersion, (*it).version);
}
} // namespace ult
} // namespace L0