mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
AubHelper: Local memory support
Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:

committed by
Compute-Runtime-Automation

parent
5e408e5e6e
commit
83db85cf86
@ -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);
|
||||
}
|
||||
|
Reference in New Issue
Block a user