mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-18 22:08:53 +08:00
82 lines
3.2 KiB
C++
82 lines
3.2 KiB
C++
/*
|
|
* Copyright (C) 2022-2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "allocation_params.h"
|
|
#include "page_info.h"
|
|
#include "shared_mem_info.h"
|
|
#include "physical_allocation_info.h"
|
|
|
|
namespace aub_stream {
|
|
|
|
struct HardwareContext;
|
|
struct CreateHardwareContext2Params;
|
|
|
|
struct AubManagerOptions {
|
|
uint32_t version{};
|
|
uint32_t productFamily{};
|
|
uint32_t devicesCount{};
|
|
uint64_t memoryBankSize{};
|
|
uint32_t stepping{};
|
|
bool localMemorySupported{};
|
|
uint32_t mode{};
|
|
uint64_t gpuAddressSpace{};
|
|
SharedMemoryInfo sharedMemoryInfo{};
|
|
bool throwOnError{};
|
|
uint64_t dataStolenMemorySize{4 * 1024 * 1024};
|
|
};
|
|
|
|
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, 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;
|
|
|
|
virtual void writeMemory2(AllocationParams allocationParams) = 0;
|
|
|
|
static AubManager *create(const struct AubManagerOptions &options);
|
|
|
|
virtual bool reservePhysicalMemory(AllocationParams allocationParams, PhysicalAllocationInfo &physicalAllocInfo) = 0;
|
|
virtual bool mapGpuVa(uint64_t gfxAddress, size_t size, PhysicalAllocationInfo physicalAllocInfo) = 0;
|
|
virtual bool mapGpuVa2(uint64_t physicalAddress, AllocationParams params) { return false; }
|
|
|
|
virtual bool reserveOnlyPhysicalSpace(AllocationParams allocationParams, PhysicalAllocationInfo &physicalAllocInfo) { return false; }
|
|
virtual bool mapSystemMemoryToPhysicalAddress(uint64_t physAddress, size_t size, size_t alignment, bool isLocalMemory, const void *p) { return false; }
|
|
virtual void *translatePhysicalAddressToSystemMemory(uint64_t physicalAddress, bool isLocalMemory) { return nullptr; }
|
|
virtual void writeMMIO(uint32_t offset, uint32_t value) {}
|
|
virtual uint32_t readMMIO(uint32_t offset) { return 0; }
|
|
virtual void writePCICFG(uint32_t offset, uint32_t value) {}
|
|
virtual uint32_t readPCICFG(uint32_t offset) { return 0; }
|
|
|
|
virtual void blockMemWritesViaTbx(bool onoff) {}
|
|
|
|
virtual bool releaseHardwareContext(HardwareContext *context) { return false; };
|
|
virtual HardwareContext *createHardwareContext2(const CreateHardwareContext2Params ¶ms, uint32_t device, uint32_t engine, uint32_t flags) { return nullptr; };
|
|
|
|
virtual void setCCSMode(uint32_t ccsCount) {}
|
|
virtual void closeSocket(void) {}
|
|
};
|
|
|
|
} // namespace aub_stream
|