mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-03 14:55:24 +08:00
a17745532c
Related-To: NEO-12323 Signed-off-by: Krzysztof Sprzaczkowski <krzysztof.sprzaczkowski@intel.com>
73 lines
2.6 KiB
C++
73 lines
2.6 KiB
C++
/*
|
|
* Copyright (C) 2020-2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "opencl/test/unit_test/fixtures/cl_device_fixture.h"
|
|
|
|
#include "shared/source/built_ins/sip.h"
|
|
|
|
#include "opencl/source/helpers/cl_gfx_core_helper.h"
|
|
#include "opencl/test/unit_test/mocks/mock_cl_device.h"
|
|
#include "opencl/test/unit_test/mocks/mock_cl_execution_environment.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace NEO {
|
|
void ClDeviceFixture::setUp() {
|
|
hardwareInfo = *defaultHwInfo;
|
|
setUpImpl(&hardwareInfo);
|
|
}
|
|
|
|
void ClDeviceFixture::setUpImpl(const NEO::HardwareInfo *hardwareInfo) {
|
|
pDevice = MockClDevice::createWithNewExecutionEnvironment<MockDevice>(hardwareInfo, rootDeviceIndex);
|
|
pDevice->getExecutionEnvironment()->rootDeviceEnvironments[pDevice->getRootDeviceIndex()]->memoryOperationsInterface = std::make_unique<MockMemoryOperations>();
|
|
ASSERT_NE(nullptr, pDevice);
|
|
pClExecutionEnvironment = static_cast<MockClExecutionEnvironment *>(pDevice->getExecutionEnvironment());
|
|
pClDevice = new MockClDevice{pDevice};
|
|
ASSERT_NE(nullptr, pClDevice);
|
|
|
|
auto &commandStreamReceiver = pDevice->getGpgpuCommandStreamReceiver();
|
|
pTagMemory = commandStreamReceiver.getTagAddress();
|
|
ASSERT_NE(nullptr, const_cast<TagAddressType *>(pTagMemory));
|
|
this->osContext = pDevice->getDefaultEngine().osContext;
|
|
if (pDevice->getPreemptionMode() == NEO::PreemptionMode::MidThread &&
|
|
!commandStreamReceiver.getPreemptionAllocation()) {
|
|
commandStreamReceiver.createPreemptionAllocation();
|
|
}
|
|
}
|
|
|
|
void ClDeviceFixture::tearDown() {
|
|
delete pClDevice;
|
|
pClDevice = nullptr;
|
|
pDevice = nullptr;
|
|
}
|
|
|
|
MockDevice *ClDeviceFixture::createWithUsDeviceId(unsigned short usDeviceId) {
|
|
hardwareInfo = *defaultHwInfo;
|
|
hardwareInfo.platform.usDeviceID = usDeviceId;
|
|
return MockDevice::createWithNewExecutionEnvironment<MockDevice>(&hardwareInfo, rootDeviceIndex);
|
|
}
|
|
|
|
const RootDeviceEnvironment &ClDeviceFixture::getRootDeviceEnvironment() const {
|
|
return pClDevice->getRootDeviceEnvironment();
|
|
}
|
|
|
|
RootDeviceEnvironment &ClDeviceFixture::getMutableRootDeviceEnvironment() {
|
|
return pClDevice->getDevice().getRootDeviceEnvironmentRef();
|
|
}
|
|
|
|
template <typename HelperType>
|
|
HelperType &ClDeviceFixture::getHelper() const {
|
|
auto &helper = pClDevice->getRootDeviceEnvironment().getHelper<HelperType>();
|
|
return helper;
|
|
}
|
|
|
|
template ProductHelper &ClDeviceFixture::getHelper() const;
|
|
template CompilerProductHelper &ClDeviceFixture::getHelper() const;
|
|
template GfxCoreHelper &ClDeviceFixture::getHelper() const;
|
|
template ClGfxCoreHelper &ClDeviceFixture::getHelper() const;
|
|
} // namespace NEO
|