intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-25 13:33:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding wsl_compute_helper

Browse Source 
Signed-off-by: Jaroslaw Chodor <jaroslaw.chodor@intel.com>
This commit is contained in:
Jaroslaw Chodor
2021-06-05 12:09:29 +02:00
committed by Compute-Runtime-Automation
parent a214497276
commit b80a51e5bf
50 changed files with 21331 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
shared/test/unit_test/CMakeLists.txt
View File

@@ -78,6 +78,12 @@ if(NOT SKIP_UNIT_TESTS)
${NEO_SOURCE_DIR}/shared/test/common/mocks/linux/mock_drm_memory_manager.cpp
${NEO_SOURCE_DIR}/shared/test/common/mocks/linux/mock_drm_memory_manager.h
)
if(NOT DISABLE_WDDM_LINUX)
target_sources(${TARGET_NAME} PRIVATE
${NEO_SHARED_DIRECTORY}/gmm_helper/windows/gmm_memory.cpp
${NEO_SHARED_DIRECTORY}/os_interface/windows/wddm/wddm_create.cpp
)
endif()
else()
target_sources(${TARGET_NAME} PRIVATE
${NEO_SOURCE_DIR}/opencl/test/unit_test/os_interface/windows/create_wddm_memory_manager.cpp

196
shared/test/unit_test/os_interface/windows/um_km_data_translator_tests.cpp
View File

@@ -1,15 +1,152 @@
/*
* Copyright (C) 2021 Intel Corporation
* Copyright (C) 2018-2021 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "shared/source/debug_settings/debug_settings_manager.h"
#include "shared/source/os_interface/windows/gdi_interface.h"
#include "shared/source/os_interface/windows/wddm/um_km_data_translator.h"
#include "shared/test/common/helpers/debug_manager_state_restore.h"
#include "shared/test/unit_test/os_interface/windows/adapter_info_tests.h"
#include "shared/wsl_compute_helper/source/gmm_resource_info_accessor.h"
#include "shared/wsl_compute_helper/source/wsl_compute_helper.h"
#include "shared/wsl_compute_helper/source/wsl_compute_helper_types_tokens_structs.h"
#include "test.h"
#ifdef _WIN32
#define EXPORT __declspec(dllexport)
#else
#define EXPORT __attribute__((__visibility__("default")))
#endif
namespace NEO {
extern const char *wslComputeHelperLibNameToLoad;
}
static const char *mockTokToStrAdapterString = "MOCK_ADAPTER_TOK_TO_STR";
static const uint16_t mockTokToStrDriverBuildNumber = 0xabcd;
static const uint32_t mockTokToStrProcessID = 0xabcdefbc;
static const char *mockStrToTokAdapterString = "MOCK_ADAPTER_STR_TO_TOK";
static const uint16_t mockStrToTokDriverBuildNumber = 0xbadc;
static const uint32_t mockStrToTokProcessID = 0xcdbaebfc;
extern "C" {
EXPORT size_t CCONV getSizeRequiredForStruct(TOK structId) {
switch (structId) {
default:
return 0;
case TOK_S_ADAPTER_INFO:
return sizeof(ADAPTER_INFO);
case TOK_S_COMMAND_BUFFER_HEADER_REC:
return sizeof(COMMAND_BUFFER_HEADER);
case TOK_S_CREATECONTEXT_PVTDATA:
return sizeof(CREATECONTEXT_PVTDATA);
case TOK_S_GMM_RESOURCE_INFO_WIN_STRUCT:
return sizeof(GmmResourceInfoWinStruct);
}
return 0;
}
EXPORT bool CCONV tokensToStruct(TOK structId, void *dst, size_t dstSizeInBytes, const TokenHeader *begin, const TokenHeader *end) {
if (dstSizeInBytes < getSizeRequiredForStruct(structId)) {
return false;
}
switch (structId) {
default:
return false;
case TOK_S_ADAPTER_INFO: {
auto adapterInfo = new (dst) ADAPTER_INFO{};
return (0 == memcpy_s(adapterInfo->AdapterString, sizeof(adapterInfo->AdapterString), mockTokToStrAdapterString, strlen(mockTokToStrAdapterString) + 1));
} break;
case TOK_S_COMMAND_BUFFER_HEADER_REC: {
auto cmdBufferHeader = new (dst) COMMAND_BUFFER_HEADER{};
cmdBufferHeader->DriverBuildNumber = mockTokToStrDriverBuildNumber;
return true;
} break;
case TOK_S_CREATECONTEXT_PVTDATA: {
auto createCtxData = new (dst) CREATECONTEXT_PVTDATA{};
createCtxData->ProcessID = mockTokToStrProcessID;
return true;
} break;
case TOK_S_GMM_RESOURCE_INFO_WIN_STRUCT: {
auto gmmResourceInfo = new (dst) GmmResourceInfoWinStruct{};
gmmResourceInfo->GmmResourceInfoCommon.pPrivateData = mockTokToStrDriverBuildNumber;
return true;
} break;
}
return true;
}
EXPORT size_t CCONV getSizeRequiredForTokens(TOK structId) {
switch (structId) {
default:
return 0;
case TOK_S_ADAPTER_INFO:
return sizeof(TOKSTR__ADAPTER_INFO);
case TOK_S_COMMAND_BUFFER_HEADER_REC:
return sizeof(TOKSTR_COMMAND_BUFFER_HEADER_REC);
case TOK_S_CREATECONTEXT_PVTDATA:
return sizeof(TOKSTR__CREATECONTEXT_PVTDATA);
case TOK_S_GMM_RESOURCE_INFO_WIN_STRUCT:
return sizeof(TOKSTR_GmmResourceInfoWinStruct);
}
return 0;
}
EXPORT bool CCONV structToTokens(TOK structId, TokenHeader *dst, size_t dstSizeInBytes, const void *src, size_t srcSizeInBytes) {
if (dstSizeInBytes < getSizeRequiredForTokens(structId)) {
return false;
}
switch (structId) {
default:
return false;
case TOK_S_ADAPTER_INFO: {
auto adapterInfo = new (dst) TOKSTR__ADAPTER_INFO{};
return (0 == memcpy_s(adapterInfo->DeviceRegistryPath.getValue<char>(), adapterInfo->DeviceRegistryPath.getValueSizeInBytes(), mockStrToTokAdapterString, strlen(mockStrToTokAdapterString) + 1));
} break;
case TOK_S_COMMAND_BUFFER_HEADER_REC: {
auto cmdBufferHeader = new (dst) TOKSTR_COMMAND_BUFFER_HEADER_REC{};
cmdBufferHeader->MonitorFenceValue.setValue(mockStrToTokDriverBuildNumber);
return true;
} break;
case TOK_S_CREATECONTEXT_PVTDATA: {
auto createCtxData = new (dst) TOKSTR__CREATECONTEXT_PVTDATA{};
createCtxData->ProcessID.setValue(mockStrToTokProcessID);
return true;
} break;
case TOK_S_GMM_RESOURCE_INFO_WIN_STRUCT: {
auto gmmResourceInfo = new (dst) TOKSTR_GmmResourceInfoWinStruct{};
gmmResourceInfo->GmmResourceInfoCommon.pPrivateData.setValue(mockStrToTokDriverBuildNumber);
return true;
} break;
}
return false;
}
EXPORT void CCONV destroyStructRepresentation(TOK structId, void *src, size_t srcSizeInBytes) {
if (srcSizeInBytes < getSizeRequiredForStruct(structId)) {
assert(false);
return;
}
switch (structId) {
default:
return;
case TOK_S_GMM_RESOURCE_INFO_WIN_STRUCT:
reinterpret_cast<GmmResourceInfoWinStruct *>(src)->~GmmResourceInfoWinStruct();
break;
}
}
EXPORT uint64_t CCONV getVersion() {
return 0;
}
}
TEST(UmKmDataTranslator, whenCreatingDefaultTranslatorThenTranslationIsDisabled) {
NEO::Gdi gdi;
auto translator = NEO::createUmKmDataTranslator(gdi, 0);
@@ -42,3 +179,60 @@ TEST(UmKmDataTranslator, whenUsingDefaultTranslatorThenTranslationIsDisabled) {
EXPECT_TRUE(ret);
EXPECT_EQ(0, memcmp(&dstCreateContextPvtData, &srcCreateContextPvtData, sizeof(CREATECONTEXT_PVTDATA)));
}
TEST(UmKmDataTranslator, givenToggledDebugKeyWhenCreatingDefaultTranslatorThenTranslationIsEnabled) {
DebugManagerStateRestore debugSettingsRestore;
NEO::DebugManager.flags.UseUmKmDataTranslator.set(true);
NEO::wslComputeHelperLibNameToLoad = "";
NEO::Gdi gdi;
auto handle = validHandle;
gdi.queryAdapterInfo.mFunc = QueryAdapterInfoMock::queryadapterinfo;
auto translator = NEO::createUmKmDataTranslator(gdi, handle);
EXPECT_TRUE(translator->enabled());
}
TEST(WslUmKmDataTranslator, whenQueryingForTranslationThenQueryIsForwardedToWslComputeHelper) {
DebugManagerStateRestore debugSettingsRestore;
NEO::DebugManager.flags.UseUmKmDataTranslator.set(true);
NEO::wslComputeHelperLibNameToLoad = "";
NEO::Gdi gdi;
auto handle = validHandle;
gdi.queryAdapterInfo.mFunc = QueryAdapterInfoMock::queryadapterinfo;
auto translator = NEO::createUmKmDataTranslator(gdi, handle);
EXPECT_EQ(sizeof(ADAPTER_INFO), translator->getSizeForAdapterInfoInternalRepresentation());
EXPECT_EQ(sizeof(COMMAND_BUFFER_HEADER), translator->getSizeForCommandBufferHeaderDataInternalRepresentation());
EXPECT_EQ(sizeof(CREATECONTEXT_PVTDATA), translator->getSizeForCreateContextDataInternalRepresentation());
ADAPTER_INFO srcAdapterInfo, dstAdapterInfo;
memset(&srcAdapterInfo, 7, sizeof(srcAdapterInfo));
auto ret = translator->translateAdapterInfoFromInternalRepresentation(dstAdapterInfo, &srcAdapterInfo, sizeof(ADAPTER_INFO));
EXPECT_TRUE(ret);
EXPECT_EQ(0, memcmp(dstAdapterInfo.DeviceRegistryPath, mockStrToTokAdapterString, strlen(mockStrToTokAdapterString) + 1));
COMMAND_BUFFER_HEADER srcCmdBufferHeader, dstCmdBufferHeader;
memset(&srcCmdBufferHeader, 7, sizeof(srcCmdBufferHeader));
ret = translator->tranlateCommandBufferHeaderDataToInternalRepresentation(&dstCmdBufferHeader, sizeof(COMMAND_BUFFER_HEADER), srcCmdBufferHeader);
EXPECT_TRUE(ret);
EXPECT_EQ(mockTokToStrDriverBuildNumber, dstCmdBufferHeader.DriverBuildNumber);
CREATECONTEXT_PVTDATA srcCreateContextPvtData, dstCreateContextPvtData;
memset(&srcCreateContextPvtData, 7, sizeof(srcCreateContextPvtData));
ret = translator->translateCreateContextDataToInternalRepresentation(&dstCreateContextPvtData, sizeof(CREATECONTEXT_PVTDATA), srcCreateContextPvtData);
EXPECT_TRUE(ret);
EXPECT_EQ(mockTokToStrProcessID, dstCreateContextPvtData.ProcessID);
auto gmmHandleAllocator = translator->createGmmHandleAllocator();
ASSERT_NE(nullptr, gmmHandleAllocator);
gmmHandleAllocator->getHandleSize();
UmKmDataTempStorage<GMM_RESOURCE_INFO> gmmResInfo;
EXPECT_EQ(sizeof(GmmResourceInfoWinStruct), gmmHandleAllocator->getHandleSize());
auto gmmResourceInfoHandle = gmmHandleAllocator->createHandle(reinterpret_cast<GMM_RESOURCE_INFO *>(gmmResInfo.data()));
ASSERT_NE(nullptr, gmmResourceInfoHandle);
EXPECT_EQ(mockTokToStrDriverBuildNumber, reinterpret_cast<GmmResourceInfoWinStruct *>(gmmResourceInfoHandle)->GmmResourceInfoCommon.pPrivateData);
gmmHandleAllocator->destroyHandle(gmmResourceInfoHandle);
}