mirror of https://github.com/alibaba/MNN.git
70 lines
2.2 KiB
C++
70 lines
2.2 KiB
C++
//
|
|
// VulkanBuffer.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2019/01/31.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include "backend/vulkan/component/VulkanBuffer.hpp"
|
|
#include <string.h>
|
|
namespace MNN {
|
|
|
|
VulkanBuffer::VulkanBuffer(const VulkanMemoryPool& pool, bool separate, size_t size, const void* hostData,
|
|
VkBufferUsageFlags usage, VkSharingMode shared, VkFlags requirements_mask)
|
|
: mPool(pool) {
|
|
MNN_ASSERT(size > 0);
|
|
mSize = size;
|
|
mShared = shared;
|
|
mBuffer = const_cast<VulkanMemoryPool&>(mPool).allocBuffer(size, usage, shared);
|
|
mUsage = usage;
|
|
|
|
VkMemoryRequirements memReq;
|
|
mPool.device().getBufferMemoryRequirements(mBuffer, memReq);
|
|
mMemory = const_cast<VulkanMemoryPool&>(mPool).allocMemory(memReq, requirements_mask, separate);
|
|
// FUNC_PRINT(mMemory->type());
|
|
auto realMem = (VulkanMemory*)mMemory.first;
|
|
|
|
if (nullptr != hostData) {
|
|
void* data = nullptr;
|
|
CALL_VK(mPool.device().mapMemory(realMem->get(), mMemory.second, size, 0 /*flag, not used*/, &data));
|
|
::memcpy(data, hostData, size);
|
|
mPool.device().unmapMemory(realMem->get());
|
|
}
|
|
CALL_VK(mPool.device().bindBufferMemory(mBuffer, realMem->get(), mMemory.second));
|
|
}
|
|
|
|
VulkanBuffer::~VulkanBuffer() {
|
|
const_cast<VulkanMemoryPool&>(mPool).returnBuffer(mBuffer, mSize, mUsage, mShared);
|
|
if (!mReleased) {
|
|
const_cast<VulkanMemoryPool&>(mPool).returnMemory(mMemory);
|
|
}
|
|
}
|
|
void* VulkanBuffer::map(int start, int size) const {
|
|
const auto& limits = mPool.device().proty().limits;
|
|
if (size < 0) {
|
|
size = mSize;
|
|
}
|
|
auto realMem = (VulkanMemory*)mMemory.first;
|
|
void* data = nullptr;
|
|
CALL_VK(mPool.device().mapMemory(realMem->get(), start + mMemory.second, size, 0, &data));
|
|
return data;
|
|
}
|
|
void VulkanBuffer::unmap() const {
|
|
auto realMem = (VulkanMemory*)mMemory.first;
|
|
mPool.device().unmapMemory(realMem->get());
|
|
}
|
|
void VulkanBuffer::release() {
|
|
if (mReleased) {
|
|
return;
|
|
}
|
|
mReleased = true;
|
|
const_cast<VulkanMemoryPool&>(mPool).returnMemory(mMemory);
|
|
}
|
|
|
|
void VulkanBuffer::flush(bool write, int start, int size) const {
|
|
// Do nothing
|
|
}
|
|
|
|
} // namespace MNN
|