Function _copyBufferToTensor

source/backend/vulkan/buffer/backend/VulkanBackend.cpp:75–85

#endif

static void _copyBufferToTensor(const Tensor* dest, const VulkanBuffer* source, size_t offset, bool half2float = false) {
    auto sourcePtr   = (const float*)source->map(offset);
    if (half2float) {
        auto dstPtr = dest->host<float>();
        auto elementCount = static_cast<size_t>(dest->elementSize());
        HALF_TO_FLOAT(reinterpret_cast<const int16_t*>(sourcePtr), dstPtr, elementCount);
    } else {
        ::memcpy(dest->host<float>(), sourcePtr, dest->usize());
    }
    source->unmap();
}

static void _copyTensorToBuffer(const Tensor* source, const VulkanBuffer* dest, size_t offset, bool float2half = false) {
    auto destPtr = reinterpret_cast<uint8_t*>(dest->map(offset));