59 lines
2.2 KiB
C++
59 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2020-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/gmm_helper/gmm_helper.h"
|
|
#include "shared/source/os_interface/os_interface.h"
|
|
#include "shared/test/common/helpers/default_hw_info.h"
|
|
#include "shared/test/common/helpers/variable_backup.h"
|
|
#include "shared/test/common/libult/linux/drm_mock.h"
|
|
#include "shared/test/common/mocks/mock_execution_environment.h"
|
|
#include "shared/test/common/test_macros/test.h"
|
|
|
|
#include "opencl/test/unit_test/linux/mock_os_layer.h"
|
|
|
|
namespace NEO {
|
|
extern GMM_INIT_IN_ARGS passedInputArgs;
|
|
extern bool copyInputArgs;
|
|
|
|
TEST(OsInterfaceTest, whenOsInterfaceSetupsGmmInputArgsThenFileDescriptorIsSetWithValueOfAdapterBdf) {
|
|
MockExecutionEnvironment executionEnvironment;
|
|
auto rootDeviceEnvironment = executionEnvironment.rootDeviceEnvironments[0].get();
|
|
auto osInterface = new OSInterface();
|
|
rootDeviceEnvironment->osInterface.reset(osInterface);
|
|
|
|
auto drm = new DrmMock(fakeFd, *rootDeviceEnvironment);
|
|
drm->setPciPath("0000:01:23.4");
|
|
EXPECT_EQ(0, drm->queryAdapterBDF());
|
|
|
|
osInterface->setDriverModel(std::unique_ptr<DriverModel>(drm));
|
|
|
|
GMM_INIT_IN_ARGS gmmInputArgs = {};
|
|
EXPECT_EQ(0u, gmmInputArgs.FileDescriptor);
|
|
osInterface->getDriverModel()->setGmmInputArgs(&gmmInputArgs);
|
|
EXPECT_NE(0u, gmmInputArgs.FileDescriptor);
|
|
|
|
ADAPTER_BDF expectedAdapterBDF{};
|
|
expectedAdapterBDF.Bus = 0x1;
|
|
expectedAdapterBDF.Device = 0x23;
|
|
expectedAdapterBDF.Function = 0x4;
|
|
EXPECT_EQ(expectedAdapterBDF.Data, gmmInputArgs.FileDescriptor);
|
|
EXPECT_EQ(GMM_CLIENT::GMM_OCL_VISTA, gmmInputArgs.ClientType);
|
|
}
|
|
|
|
TEST(GmmHelperTest, whenCreateGmmHelperWithoutOsInterfaceThenPassedFileDescriptorIsZeroed) {
|
|
MockExecutionEnvironment executionEnvironment{};
|
|
VariableBackup<decltype(passedInputArgs)> passedInputArgsBackup(&passedInputArgs);
|
|
VariableBackup<decltype(copyInputArgs)> copyInputArgsBackup(©InputArgs, true);
|
|
|
|
uint32_t expectedFileDescriptor = 0u;
|
|
|
|
auto gmmHelper = std::make_unique<GmmHelper>(*executionEnvironment.rootDeviceEnvironments[0]);
|
|
EXPECT_EQ(expectedFileDescriptor, passedInputArgs.FileDescriptor);
|
|
EXPECT_EQ(GMM_CLIENT::GMM_OCL_VISTA, passedInputArgs.ClientType);
|
|
}
|
|
} // namespace NEO
|