AubHelper: Local memory support

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2021-07-05 14:52:03 +00:00
committed by Compute-Runtime-Automation
parent 5e408e5e6e
commit 83db85cf86
10 changed files with 249 additions and 158 deletions

View File

@ -6,6 +6,7 @@
*/
#include "shared/source/command_stream/tbx_command_stream_receiver_hw.h"
#include "shared/source/tbx/tbx_proto.h"
#include "shared/test/common/mocks/mock_tbx_sockets.h"
#include "shared/test/common/mocks/mock_tbx_stream.h"
@ -13,16 +14,26 @@
using namespace NEO;
TEST(TbxStreamTests, givenTbxStreamWhenWriteMemoryIsCalledThenTypeIsZero) {
TEST(TbxStreamTests, givenTbxStreamWhenWriteMemoryIsCalledThenMemTypeIsSetCorrectly) {
std::unique_ptr<TbxCommandStreamReceiver::TbxStream> mockTbxStream(new MockTbxStream());
MockTbxStream *mockTbxStreamPtr = static_cast<MockTbxStream *>(mockTbxStream.get());
MockTbxSockets *mockTbxSocket = new MockTbxSockets();
mockTbxStreamPtr->socket = mockTbxSocket;
mockTbxStream->writeMemory(0, nullptr, 0, 0, 0);
EXPECT_EQ(0u, mockTbxSocket->typeCapturedFromWriteMemory);
uint32_t addressSpace = AubMemDump::AddressSpaceValues::TraceLocal;
mockTbxStream->writeMemory(0, nullptr, 0, addressSpace, 0);
EXPECT_EQ(mem_types::MEM_TYPE_LOCALMEM, mockTbxSocket->typeCapturedFromWriteMemory);
mockTbxStream->writePTE(0, 0, 0);
EXPECT_EQ(0u, mockTbxSocket->typeCapturedFromWriteMemory);
addressSpace = AubMemDump::AddressSpaceValues::TraceNonlocal;
mockTbxStream->writeMemory(0, nullptr, 0, addressSpace, 0);
EXPECT_EQ(mem_types::MEM_TYPE_SYSTEM, mockTbxSocket->typeCapturedFromWriteMemory);
addressSpace = AubMemDump::AddressSpaceValues::TraceLocal;
mockTbxStream->writePTE(0, 0, addressSpace);
EXPECT_EQ(mem_types::MEM_TYPE_LOCALMEM, mockTbxSocket->typeCapturedFromWriteMemory);
addressSpace = AubMemDump::AddressSpaceValues::TraceNonlocal;
mockTbxStream->writePTE(0, 0, addressSpace);
EXPECT_EQ(mem_types::MEM_TYPE_SYSTEM, mockTbxSocket->typeCapturedFromWriteMemory);
}