154 lines
7.1 KiB
C++
154 lines
7.1 KiB
C++
/*
|
|
* Copyright (C) 2018-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "core/helpers/basic_math.h"
|
|
#include "core/unit_tests/helpers/debug_manager_state_restore.h"
|
|
#include "runtime/helpers/hw_info.h"
|
|
#include "runtime/helpers/options.h"
|
|
#include "runtime/os_interface/debug_settings_manager.h"
|
|
#include "unit_tests/mocks/mock_aub_center.h"
|
|
#include "unit_tests/mocks/mock_aub_manager.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "third_party/aub_stream/headers/aubstream.h"
|
|
|
|
using namespace NEO;
|
|
|
|
TEST(AubCenter, GivenUseAubStreamDebugVariableNotSetWhenAubCenterIsCreatedThenAubCenterDoesNotCreateAubManager) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(false);
|
|
|
|
MockAubCenter aubCenter(platformDevices[0], false, "", CommandStreamReceiverType::CSR_AUB);
|
|
EXPECT_EQ(nullptr, aubCenter.aubManager.get());
|
|
}
|
|
|
|
TEST(AubCenter, GivenUseAubStreamDebugVariableSetWhenAubCenterIsCreatedThenCreateAubManagerWithCorrectParameters) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(false);
|
|
|
|
MockAubManager *mockAubManager = new MockAubManager(platformDevices[0]->platform.eProductFamily, 4, 8 * MB, true, aub_stream::mode::aubFile);
|
|
MockAubCenter mockAubCenter(platformDevices[0], false, "", CommandStreamReceiverType::CSR_AUB);
|
|
mockAubCenter.aubManager = std::unique_ptr<MockAubManager>(mockAubManager);
|
|
|
|
EXPECT_EQ(platformDevices[0]->platform.eProductFamily, mockAubManager->mockAubManagerParams.productFamily);
|
|
EXPECT_EQ(4, mockAubManager->mockAubManagerParams.devicesCount);
|
|
EXPECT_EQ(8 * MB, mockAubManager->mockAubManagerParams.memoryBankSize);
|
|
EXPECT_EQ(true, mockAubManager->mockAubManagerParams.localMemorySupported);
|
|
EXPECT_EQ(aub_stream::mode::aubFile, mockAubManager->mockAubManagerParams.streamMode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenDefaultSetCommandStreamReceiverFlagAndAubFileNameWhenGettingAubStreamModeThenModeAubFileIsReturned) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
|
|
std::string aubFile("test.aub");
|
|
auto mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_AUB);
|
|
|
|
EXPECT_EQ(aub_stream::mode::aubFile, mode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenCsrHwAndEmptyAubFileNameWhenGettingAubStreamModeThenModeAubFileIsReturned) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
|
|
std::string aubFile("");
|
|
auto mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_HW);
|
|
|
|
EXPECT_EQ(aub_stream::mode::aubFile, mode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenCsrHwAndNotEmptyAubFileNameWhenGettingAubStreamModeThenModeAubFileIsReturned) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
|
|
std::string aubFile("test.aub");
|
|
auto mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_HW);
|
|
|
|
EXPECT_EQ(aub_stream::mode::aubFile, mode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenCsrTypeWhenGettingAubStreamModeThenCorrectModeIsReturned) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
|
|
std::string aubFile("test.aub");
|
|
|
|
auto mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_AUB);
|
|
EXPECT_EQ(aub_stream::mode::aubFile, mode);
|
|
|
|
mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_TBX);
|
|
EXPECT_EQ(aub_stream::mode::tbx, mode);
|
|
|
|
mode = AubCenter::getAubStreamMode(aubFile, CommandStreamReceiverType::CSR_TBX_WITH_AUB);
|
|
EXPECT_EQ(aub_stream::mode::aubFileAndTbx, mode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenSetCommandStreamReceiverFlagEqualDefaultHwWhenAubManagerIsCreatedThenCsrTypeDefinesAubStreamMode) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
DebugManager.flags.SetCommandStreamReceiver.set(-1);
|
|
|
|
std::vector<CommandStreamReceiverType> aubTypes = {CommandStreamReceiverType::CSR_HW,
|
|
CommandStreamReceiverType::CSR_HW_WITH_AUB,
|
|
CommandStreamReceiverType::CSR_AUB};
|
|
|
|
for (auto type : aubTypes) {
|
|
MockAubCenter aubCenter(platformDevices[0], true, "test", type);
|
|
EXPECT_EQ(aub_stream::mode::aubFile, aubCenter.aubStreamMode);
|
|
}
|
|
|
|
MockAubCenter aubCenter2(platformDevices[0], true, "", CommandStreamReceiverType::CSR_TBX);
|
|
EXPECT_EQ(aub_stream::mode::tbx, aubCenter2.aubStreamMode);
|
|
|
|
MockAubCenter aubCenter3(platformDevices[0], true, "", CommandStreamReceiverType::CSR_TBX_WITH_AUB);
|
|
EXPECT_EQ(aub_stream::mode::aubFileAndTbx, aubCenter3.aubStreamMode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenSetCommandStreamReceiverFlagSetWhenAubManagerIsCreatedThenDebugFlagDefinesAubStreamMode) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.UseAubStream.set(true);
|
|
|
|
DebugManager.flags.SetCommandStreamReceiver.set(CommandStreamReceiverType::CSR_TBX);
|
|
|
|
MockAubCenter aubCenter(platformDevices[0], true, "", CommandStreamReceiverType::CSR_AUB);
|
|
EXPECT_EQ(aub_stream::mode::tbx, aubCenter.aubStreamMode);
|
|
|
|
DebugManager.flags.SetCommandStreamReceiver.set(CommandStreamReceiverType::CSR_TBX_WITH_AUB);
|
|
|
|
MockAubCenter aubCenter2(platformDevices[0], true, "", CommandStreamReceiverType::CSR_AUB);
|
|
EXPECT_EQ(aub_stream::mode::aubFileAndTbx, aubCenter2.aubStreamMode);
|
|
}
|
|
|
|
TEST(AubCenter, GivenAubCenterInSubCaptureModeWhenItIsCreatedWithoutDebugFilterSettingsThenItInitializesSubCaptureFiltersWithDefaults) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.AUBDumpSubCaptureMode.set(static_cast<int32_t>(AubSubCaptureManager::SubCaptureMode::Filter));
|
|
|
|
MockAubCenter aubCenter(platformDevices[0], false, "", CommandStreamReceiverType::CSR_AUB);
|
|
auto subCaptureCommon = aubCenter.getSubCaptureCommon();
|
|
EXPECT_NE(nullptr, subCaptureCommon);
|
|
|
|
EXPECT_EQ(0u, subCaptureCommon->subCaptureFilter.dumpKernelStartIdx);
|
|
EXPECT_EQ(static_cast<uint32_t>(-1), subCaptureCommon->subCaptureFilter.dumpKernelEndIdx);
|
|
EXPECT_STREQ("", subCaptureCommon->subCaptureFilter.dumpKernelName.c_str());
|
|
}
|
|
|
|
TEST(AubCenter, GivenAubCenterInSubCaptureModeWhenItIsCreatedWithDebugFilterSettingsThenItInitializesSubCaptureFiltersWithDebugFilterSettings) {
|
|
DebugManagerStateRestore restorer;
|
|
DebugManager.flags.AUBDumpSubCaptureMode.set(static_cast<int32_t>(AubSubCaptureManager::SubCaptureMode::Filter));
|
|
DebugManager.flags.AUBDumpFilterKernelStartIdx.set(10);
|
|
DebugManager.flags.AUBDumpFilterKernelEndIdx.set(100);
|
|
DebugManager.flags.AUBDumpFilterKernelName.set("kernel_name");
|
|
|
|
MockAubCenter aubCenter(platformDevices[0], false, "", CommandStreamReceiverType::CSR_AUB);
|
|
auto subCaptureCommon = aubCenter.getSubCaptureCommon();
|
|
EXPECT_NE(nullptr, subCaptureCommon);
|
|
|
|
EXPECT_EQ(static_cast<uint32_t>(DebugManager.flags.AUBDumpFilterKernelStartIdx.get()), subCaptureCommon->subCaptureFilter.dumpKernelStartIdx);
|
|
EXPECT_EQ(static_cast<uint32_t>(DebugManager.flags.AUBDumpFilterKernelEndIdx.get()), subCaptureCommon->subCaptureFilter.dumpKernelEndIdx);
|
|
EXPECT_STREQ(DebugManager.flags.AUBDumpFilterKernelName.get().c_str(), subCaptureCommon->subCaptureFilter.dumpKernelName.c_str());
|
|
}
|