95 lines
2.7 KiB
C++
95 lines
2.7 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include <atomic>
|
|
#include <cstdlib>
|
|
#include <errno.h>
|
|
#include <set>
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
|
|
struct drm_i915_gem_exec_object2;
|
|
struct drm_i915_gem_relocation_entry;
|
|
|
|
namespace NEO {
|
|
|
|
class DrmMemoryManager;
|
|
class Drm;
|
|
|
|
enum StorageAllocatorType {
|
|
MMAP_ALLOCATOR,
|
|
BIT32_ALLOCATOR_EXTERNAL,
|
|
BIT32_ALLOCATOR_INTERNAL,
|
|
MALLOC_ALLOCATOR,
|
|
EXTERNAL_ALLOCATOR,
|
|
INTERNAL_ALLOCATOR_WITH_DYNAMIC_BITRANGE,
|
|
UNKNOWN_ALLOCATOR
|
|
};
|
|
|
|
class BufferObject {
|
|
friend DrmMemoryManager;
|
|
|
|
public:
|
|
using ResidencyVector = std::vector<BufferObject *>;
|
|
MOCKABLE_VIRTUAL ~BufferObject(){};
|
|
|
|
bool setTiling(uint32_t mode, uint32_t stride);
|
|
|
|
MOCKABLE_VIRTUAL int pin(BufferObject *const boToPin[], size_t numberOfBos, uint32_t drmContextId);
|
|
|
|
int exec(uint32_t used, size_t startOffset, unsigned int flags, bool requiresCoherency, uint32_t drmContextId, ResidencyVector &residency, drm_i915_gem_exec_object2 *execObjectsStorage);
|
|
|
|
int wait(int64_t timeoutNs);
|
|
bool close();
|
|
|
|
inline void reference() {
|
|
this->refCount++;
|
|
}
|
|
uint32_t getRefCount() const;
|
|
|
|
bool peekIsAllocated() const { return isAllocated; }
|
|
size_t peekSize() const { return size; }
|
|
int peekHandle() const { return handle; }
|
|
uint64_t peekAddress() const { return gpuAddress; }
|
|
void setAddress(uint64_t address) { this->gpuAddress = address; }
|
|
void *peekLockedAddress() const { return lockedAddress; }
|
|
void setLockedAddress(void *cpuAddress) { this->lockedAddress = cpuAddress; }
|
|
void setUnmapSize(uint64_t unmapSize) { this->unmapSize = unmapSize; }
|
|
uint64_t peekUnmapSize() const { return unmapSize; }
|
|
StorageAllocatorType peekAllocationType() const { return storageAllocatorType; }
|
|
void setAllocationType(StorageAllocatorType allocatorType) { this->storageAllocatorType = allocatorType; }
|
|
bool peekIsReusableAllocation() { return this->isReused; }
|
|
|
|
protected:
|
|
BufferObject(Drm *drm, int handle, bool isAllocated);
|
|
|
|
Drm *drm;
|
|
|
|
std::atomic<uint32_t> refCount;
|
|
|
|
int handle; // i915 gem object handle
|
|
bool isReused;
|
|
|
|
//Tiling
|
|
uint32_t tiling_mode;
|
|
uint32_t stride;
|
|
|
|
MOCKABLE_VIRTUAL void fillExecObject(drm_i915_gem_exec_object2 &execObject, uint32_t drmContextId);
|
|
void processRelocs(int &idx, uint32_t drmContextId, ResidencyVector &residency, drm_i915_gem_exec_object2 *execObjectsStorage);
|
|
|
|
uint64_t gpuAddress = 0llu;
|
|
|
|
uint64_t size;
|
|
void *lockedAddress; // CPU side virtual address
|
|
|
|
bool isAllocated = false;
|
|
uint64_t unmapSize = 0;
|
|
StorageAllocatorType storageAllocatorType = UNKNOWN_ALLOCATOR;
|
|
};
|
|
} // namespace NEO
|