48 lines
1.2 KiB
C++
48 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "opencl/source/helpers/properties_helper.h"
|
|
|
|
#include <mutex>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
namespace NEO {
|
|
|
|
class MapOperationsHandler {
|
|
public:
|
|
virtual ~MapOperationsHandler() = default;
|
|
|
|
bool add(void *ptr, size_t ptrLength, cl_map_flags &mapFlags, MemObjSizeArray &size, MemObjOffsetArray &offset, uint32_t mipLevel, GraphicsAllocation *graphicsAllocation);
|
|
void remove(void *mappedPtr);
|
|
bool find(void *mappedPtr, MapInfo &outMapInfo);
|
|
bool findInfoForHostPtr(const void *ptr, size_t size, MapInfo &outMapInfo);
|
|
size_t size() const;
|
|
|
|
protected:
|
|
bool isOverlapping(MapInfo &inputMapInfo);
|
|
std::vector<MapInfo> mappedPointers;
|
|
mutable std::mutex mtx;
|
|
};
|
|
|
|
class MapOperationsStorage {
|
|
public:
|
|
using HandlersMap = std::unordered_map<cl_mem, MapOperationsHandler>;
|
|
|
|
MapOperationsHandler &getHandler(cl_mem memObj);
|
|
MapOperationsHandler *getHandlerIfExists(cl_mem memObj);
|
|
bool getInfoForHostPtr(const void *ptr, size_t size, MapInfo &outInfo);
|
|
void removeHandler(cl_mem memObj);
|
|
|
|
protected:
|
|
std::mutex mutex;
|
|
HandlersMap handlers{};
|
|
};
|
|
|
|
} // namespace NEO
|