56 lines
2.2 KiB
C++
56 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/aub_mem_dump/aub_mem_dump.h"
|
|
#include "shared/source/helpers/common_types.h"
|
|
|
|
namespace NEO {
|
|
class CommandStreamReceiver;
|
|
class TbxSockets;
|
|
class ExecutionEnvironment;
|
|
|
|
class TbxStream : public AubMemDump::AubStream {
|
|
protected:
|
|
TbxSockets *socket = nullptr;
|
|
|
|
public:
|
|
TbxStream();
|
|
~TbxStream() override;
|
|
|
|
TbxStream(const TbxStream &) = delete;
|
|
TbxStream &operator=(const TbxStream &) = delete;
|
|
|
|
void open(const char *options) override;
|
|
void close() override;
|
|
bool init(uint32_t stepping, uint32_t device) override;
|
|
void writeMemory(uint64_t physAddress, const void *memory, size_t size, uint32_t addressSpace, uint32_t hint) override;
|
|
void writeMemoryWriteHeader(uint64_t physAddress, size_t size, uint32_t addressSpace, uint32_t hint) override;
|
|
void writeGTT(uint32_t gttOffset, uint64_t entry) override;
|
|
void writePTE(uint64_t physAddress, uint64_t entry, uint32_t addressSpace) override;
|
|
void writeMMIOImpl(uint32_t offset, uint32_t value) override;
|
|
void registerPoll(uint32_t registerOffset, uint32_t mask, uint32_t value, bool pollNotEqual, uint32_t timeoutAction) override;
|
|
void readMemory(uint64_t physAddress, void *memory, size_t size);
|
|
};
|
|
|
|
struct TbxCommandStreamReceiver {
|
|
static CommandStreamReceiver *create(const std::string &baseName,
|
|
bool withAubDump,
|
|
ExecutionEnvironment &executionEnvironment,
|
|
uint32_t rootDeviceIndex,
|
|
const DeviceBitfield deviceBitfield);
|
|
|
|
using TbxStream = NEO::TbxStream;
|
|
};
|
|
|
|
typedef CommandStreamReceiver *(*TbxCommandStreamReceiverCreateFunc)(const std::string &baseName,
|
|
bool withAubDump,
|
|
ExecutionEnvironment &executionEnvironment,
|
|
uint32_t rootDeviceIndex,
|
|
const DeviceBitfield deviceBitfield);
|
|
} // namespace NEO
|