From ebcbbcf8fe33466c81eec37b804e1307940820da Mon Sep 17 00:00:00 2001 From: Bartosz Dunajski Date: Fri, 26 Jul 2024 08:58:19 +0000 Subject: [PATCH] fix: add missing api exports Signed-off-by: Bartosz Dunajski --- .../driver/extension_function_address.cpp | 7 ++++ .../unit_tests/sources/driver/test_driver.cpp | 32 +++++++++++++------ 2 files changed, 29 insertions(+), 10 deletions(-) diff --git a/level_zero/core/source/driver/extension_function_address.cpp b/level_zero/core/source/driver/extension_function_address.cpp index 37ca567ee2..50ef7a10aa 100644 --- a/level_zero/core/source/driver/extension_function_address.cpp +++ b/level_zero/core/source/driver/extension_function_address.cpp @@ -8,6 +8,7 @@ #include "level_zero/core/source/driver/extension_function_address.h" #include "level_zero/api/driver_experimental/public/zex_api.h" +#include "level_zero/api/driver_experimental/public/zex_context.h" #include "level_zero/api/extensions/public/ze_exp_ext.h" #include "level_zero/include/ze_intel_gpu.h" @@ -43,6 +44,12 @@ void *ExtensionFunctionAddressHelper::getExtensionFunctionAddress(const std::str RETURN_FUNC_PTR_IF_EXIST(zeImageGetDeviceOffsetExp); RETURN_FUNC_PTR_IF_EXIST(zexDeviceGetConcurrentMetricGroups); RETURN_FUNC_PTR_IF_EXIST(zeIntelGetDriverVersionString); + + RETURN_FUNC_PTR_IF_EXIST(zeIntelMediaCommunicationCreate); + RETURN_FUNC_PTR_IF_EXIST(zeIntelMediaCommunicationDestroy); + + RETURN_FUNC_PTR_IF_EXIST(zexIntelAllocateNetworkInterrupt); + RETURN_FUNC_PTR_IF_EXIST(zexIntelReleaseNetworkInterrupt); #undef RETURN_FUNC_PTR_IF_EXIST return ExtensionFunctionAddressHelper::getAdditionalExtensionFunctionAddress(functionName); diff --git a/level_zero/core/test/unit_tests/sources/driver/test_driver.cpp b/level_zero/core/test/unit_tests/sources/driver/test_driver.cpp index d7a488dbfa..941603e07f 100644 --- a/level_zero/core/test/unit_tests/sources/driver/test_driver.cpp +++ b/level_zero/core/test/unit_tests/sources/driver/test_driver.cpp @@ -29,6 +29,7 @@ #include "shared/test/common/test_macros/hw_test.h" #include "level_zero/api/driver_experimental/public/zex_api.h" +#include "level_zero/api/driver_experimental/public/zex_context.h" #include "level_zero/api/driver_experimental/public/zex_driver.h" #include "level_zero/core/source/builtin/builtin_functions_lib_impl.h" #include "level_zero/core/source/driver/driver_handle_imp.h" @@ -1323,28 +1324,39 @@ TEST_F(DriverExperimentalApiTest, whenRetrievingApiFunctionThenExpectProperPoint decltype(&zexDriverGetHostPointerBaseAddress) expectedGet = L0::zexDriverGetHostPointerBaseAddress; decltype(&zexKernelGetBaseAddress) expectedKernelGetBaseAddress = L0::zexKernelGetBaseAddress; decltype(&zeIntelGetDriverVersionString) expectedIntelGetDriverVersionString = zeIntelGetDriverVersionString; + decltype(&zeIntelMediaCommunicationCreate) expectedIntelMediaCommunicationCreate = L0::zeIntelMediaCommunicationCreate; + decltype(&zeIntelMediaCommunicationDestroy) expectedIntelMediaCommunicationDestroy = L0::zeIntelMediaCommunicationDestroy; + decltype(&zexIntelAllocateNetworkInterrupt) expectedIntelAllocateNetworkInterrupt = L0::zexIntelAllocateNetworkInterrupt; + decltype(&zexIntelReleaseNetworkInterrupt) expectedIntelReleaseNetworkInterrupt = L0::zexIntelReleaseNetworkInterrupt; void *funPtr = nullptr; - auto result = zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverImportExternalPointer", &funPtr); - EXPECT_EQ(ZE_RESULT_SUCCESS, result); + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverImportExternalPointer", &funPtr)); EXPECT_EQ(expectedImport, reinterpret_cast(funPtr)); - result = zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverReleaseImportedPointer", &funPtr); - EXPECT_EQ(ZE_RESULT_SUCCESS, result); + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverReleaseImportedPointer", &funPtr)); EXPECT_EQ(expectedRelease, reinterpret_cast(funPtr)); - result = zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverGetHostPointerBaseAddress", &funPtr); - EXPECT_EQ(ZE_RESULT_SUCCESS, result); + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexDriverGetHostPointerBaseAddress", &funPtr)); EXPECT_EQ(expectedGet, reinterpret_cast(funPtr)); - result = zeDriverGetExtensionFunctionAddress(driverHandle, "zexKernelGetBaseAddress", &funPtr); - EXPECT_EQ(ZE_RESULT_SUCCESS, result); + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexKernelGetBaseAddress", &funPtr)); EXPECT_EQ(expectedKernelGetBaseAddress, reinterpret_cast(funPtr)); - result = zeDriverGetExtensionFunctionAddress(driverHandle, "zeIntelGetDriverVersionString", &funPtr); - EXPECT_EQ(ZE_RESULT_SUCCESS, result); + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zeIntelGetDriverVersionString", &funPtr)); EXPECT_EQ(expectedIntelGetDriverVersionString, reinterpret_cast(funPtr)); + + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zeIntelMediaCommunicationCreate", &funPtr)); + EXPECT_EQ(expectedIntelMediaCommunicationCreate, reinterpret_cast(funPtr)); + + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zeIntelMediaCommunicationDestroy", &funPtr)); + EXPECT_EQ(expectedIntelMediaCommunicationDestroy, reinterpret_cast(funPtr)); + + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexIntelAllocateNetworkInterrupt", &funPtr)); + EXPECT_EQ(expectedIntelAllocateNetworkInterrupt, reinterpret_cast(funPtr)); + + EXPECT_EQ(ZE_RESULT_SUCCESS, zeDriverGetExtensionFunctionAddress(driverHandle, "zexIntelReleaseNetworkInterrupt", &funPtr)); + EXPECT_EQ(expectedIntelReleaseNetworkInterrupt, reinterpret_cast(funPtr)); } TEST_F(DriverExperimentalApiTest, givenHostPointerApiExistWhenImportingPtrThenExpectProperBehavior) {