Files
compute-runtime/third_party/aub_stream/headers/aub_manager.h
Klein, Yaniv 340e80bb1a added support for 2 stages write memory of aub_manager
1. write ppgtt pages only by calling write memory with pagesOnly set to true
2. write the actual memory pages by calling writePhysicalMemoryPages

Change-Id: Id3766704e4b8f84f77336f78da6b9270bfd44b70
2019-11-13 11:45:08 +01:00

42 lines
1.4 KiB
C++

/*
* Copyright (C) 2018-2019 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include <cstdint>
#include <string>
#include <vector>
#include "page_info.h"
namespace aub_stream {
struct HardwareContext;
class AubManager {
public:
virtual ~AubManager() = default;
virtual HardwareContext *createHardwareContext(uint32_t device, uint32_t engine, uint32_t flags) = 0;
virtual void open(const std::string &aubFileName) = 0;
virtual void close() = 0;
virtual bool isOpen() = 0;
virtual const std::string getFileName() = 0;
virtual void pause(bool onoff) = 0;
virtual void addComment(const char *message) = 0;
virtual void writeMemory(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks,int hint, size_t pageSize) = 0;
virtual void writePageTableEntries(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks, int hint,
std::vector<PageInfo> &lastLevelPages, size_t pageSize) = 0;
virtual void writePhysicalMemoryPages(const void *memory, std::vector<PageInfo> &pages, size_t size, int hint) = 0;
virtual void freeMemory(uint64_t gfxAddress, size_t size) = 0;
static AubManager *create(uint32_t productFamily, uint32_t devicesCount, uint64_t memoryBankSize, bool localMemorySupported, uint32_t streamMode);
};
} // namespace aub_stream