132 lines
5.2 KiB
C++
132 lines
5.2 KiB
C++
/*
|
|
* Copyright (C) 2017-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/aub_mem_dump/aub_mem_dump.h"
|
|
#include "shared/source/aub_mem_dump/page_table_entry_bits.h"
|
|
#include "shared/source/helpers/hw_helper.h"
|
|
#include "shared/source/os_interface/os_interface.h"
|
|
#include "shared/test/unit_test/mocks/mock_device.h"
|
|
#include "shared/test/unit_test/tests_configuration.h"
|
|
|
|
#include "opencl/source/command_stream/aub_command_stream_receiver_hw.h"
|
|
#include "opencl/source/command_stream/command_stream_receiver_with_aub_dump.h"
|
|
#include "opencl/source/command_stream/tbx_command_stream_receiver_hw.h"
|
|
#include "opencl/source/platform/platform.h"
|
|
#include "opencl/test/unit_test/command_queue/command_queue_fixture.h"
|
|
#include "opencl/test/unit_test/mocks/mock_cl_device.h"
|
|
#include "opencl/test/unit_test/mocks/mock_memory_operations_handler.h"
|
|
#include "opencl/test/unit_test/mocks/mock_platform.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <sstream>
|
|
|
|
namespace NEO {
|
|
|
|
class AUBFixture : public CommandQueueHwFixture {
|
|
public:
|
|
void SetUp(const HardwareInfo *hardwareInfo) {
|
|
const HardwareInfo &hwInfo = hardwareInfo ? *hardwareInfo : *defaultHwInfo;
|
|
|
|
auto &hwHelper = HwHelper::get(hwInfo.platform.eRenderCoreFamily);
|
|
auto engineType = getChosenEngineType(hwInfo);
|
|
|
|
const ::testing::TestInfo *const testInfo = ::testing::UnitTest::GetInstance()->current_test_info();
|
|
std::stringstream strfilename;
|
|
strfilename << testInfo->test_case_name() << "_" << testInfo->name() << "_" << hwHelper.getCsTraits(engineType).name;
|
|
|
|
executionEnvironment = platform()->peekExecutionEnvironment();
|
|
executionEnvironment->prepareRootDeviceEnvironments(1u);
|
|
executionEnvironment->rootDeviceEnvironments[0]->setHwInfo(&hwInfo);
|
|
executionEnvironment->rootDeviceEnvironments[0]->memoryOperationsInterface = std::make_unique<MockMemoryOperationsHandler>();
|
|
|
|
device = std::make_unique<MockClDevice>(MockDevice::create<MockDevice>(executionEnvironment, rootDeviceIndex));
|
|
|
|
if (testMode == TestMode::AubTestsWithTbx) {
|
|
this->csr = TbxCommandStreamReceiver::create(strfilename.str(), true, *executionEnvironment, 0, device->getDeviceBitfield());
|
|
} else {
|
|
this->csr = AUBCommandStreamReceiver::create(strfilename.str(), true, *executionEnvironment, 0, device->getDeviceBitfield());
|
|
}
|
|
|
|
device->resetCommandStreamReceiver(this->csr);
|
|
|
|
CommandQueueHwFixture::SetUp(AUBFixture::device.get(), cl_command_queue_properties(0));
|
|
}
|
|
void TearDown() override {
|
|
CommandQueueHwFixture::TearDown();
|
|
}
|
|
|
|
GraphicsAllocation *createHostPtrAllocationFromSvmPtr(void *svmPtr, size_t size);
|
|
|
|
template <typename FamilyType>
|
|
CommandStreamReceiverSimulatedCommonHw<FamilyType> *getSimulatedCsr() const {
|
|
return static_cast<CommandStreamReceiverSimulatedCommonHw<FamilyType> *>(csr);
|
|
}
|
|
|
|
template <typename FamilyType>
|
|
void expectMemory(void *gfxAddress, const void *srcAddress, size_t length) {
|
|
CommandStreamReceiverSimulatedCommonHw<FamilyType> *csrSimulated = getSimulatedCsr<FamilyType>();
|
|
|
|
if (testMode == TestMode::AubTestsWithTbx) {
|
|
auto tbxCsr = csrSimulated;
|
|
EXPECT_TRUE(tbxCsr->expectMemoryEqual(gfxAddress, srcAddress, length));
|
|
csrSimulated = static_cast<CommandStreamReceiverSimulatedCommonHw<FamilyType> *>(
|
|
static_cast<CommandStreamReceiverWithAUBDump<TbxCommandStreamReceiverHw<FamilyType>> *>(csr)->aubCSR.get());
|
|
}
|
|
|
|
if (csrSimulated) {
|
|
csrSimulated->expectMemoryEqual(gfxAddress, srcAddress, length);
|
|
}
|
|
}
|
|
|
|
template <typename FamilyType>
|
|
void expectNotEqualMemory(void *gfxAddress, const void *srcAddress, size_t length) {
|
|
CommandStreamReceiverSimulatedCommonHw<FamilyType> *csrSimulated = getSimulatedCsr<FamilyType>();
|
|
|
|
if (testMode == TestMode::AubTestsWithTbx) {
|
|
auto tbxCsr = csrSimulated;
|
|
EXPECT_TRUE(tbxCsr->expectMemoryNotEqual(gfxAddress, srcAddress, length));
|
|
csrSimulated = static_cast<CommandStreamReceiverSimulatedCommonHw<FamilyType> *>(
|
|
static_cast<CommandStreamReceiverWithAUBDump<TbxCommandStreamReceiverHw<FamilyType>> *>(csr)->aubCSR.get());
|
|
}
|
|
|
|
if (csrSimulated) {
|
|
csrSimulated->expectMemoryNotEqual(gfxAddress, srcAddress, length);
|
|
}
|
|
}
|
|
|
|
static void *getGpuPointer(GraphicsAllocation *allocation) {
|
|
return reinterpret_cast<void *>(allocation->getGpuAddress());
|
|
}
|
|
|
|
const uint32_t rootDeviceIndex = 0;
|
|
CommandStreamReceiver *csr = nullptr;
|
|
volatile uint32_t *pTagMemory = nullptr;
|
|
std::unique_ptr<MockClDevice> device;
|
|
|
|
ExecutionEnvironment *executionEnvironment;
|
|
|
|
private:
|
|
using CommandQueueHwFixture::SetUp;
|
|
}; // namespace NEO
|
|
|
|
template <typename KernelFixture>
|
|
struct KernelAUBFixture : public AUBFixture,
|
|
public KernelFixture {
|
|
void SetUp() override {
|
|
AUBFixture::SetUp(nullptr);
|
|
KernelFixture::SetUp(device.get(), context);
|
|
}
|
|
|
|
void TearDown() override {
|
|
KernelFixture::TearDown();
|
|
AUBFixture::TearDown();
|
|
}
|
|
};
|
|
} // namespace NEO
|