Method onEncodeFast

source/backend/vulkan/buffer/execution/VulkanRaster.cpp:66–113

}

void VulkanRaster::onEncodeFast(const Tensor* input, const Tensor* output, const VulkanCommandPool::Buffer *cmdBuffer, bool zero) {
    auto des = TensorUtils::getDescribe(input);
    auto vkBn = static_cast<VulkanBackend*>(backend());
    auto dstTensor = vkBn->getTensorBuffer(output);
    auto dstTensorSize = vkBn->getTensorSize(output);
    if (zero) {
        vkCmdFillBuffer(cmdBuffer->get(), dstTensor.first->buffer(), dstTensor.second, dstTensorSize, 0);
        cmdBuffer->barrierSource(dstTensor.first->buffer(), dstTensor.second, dstTensorSize, VulkanCommandPool::Buffer::WRITE_WRITE);
    }
    
    std::string pKey = "glsl_blit_C4_";
    if (output->getType().code == halide_type_float && vkBn->useFP16()) {
        pKey += "FP16_";
    }
    pKey += "comp";
    auto blitPipeline = vkBn->getPipeline(pKey, {
        VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
        VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
        VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER
    });

    for (int i=0; i< des->regions.size(); ++i) {
        auto& slice = des->regions[i];
        Tensor::InsideDescribe::Region newRegion;
        OpCommonUtils::turnToPackRegion(slice, newRegion, output, 4, true);
        // TODO: Find better way
        newRegion.dst.offset /= 4;
        newRegion.src.offset /= 4;
        SamplerInfo info;
        writeSamplerInfo(info, newRegion);
        auto total = info.size[0] * info.size[1] * info.size[2];
        auto group = UP_DIV(total, 256);
        std::shared_ptr<VulkanLayout::DescriptorSet> describe(blitPipeline->createSet());
        std::shared_ptr<VulkanBuffer> uniform = vkBn->allocUniform();
        ::memcpy(uniform->map(), &info, sizeof(SamplerInfo));
        uniform->unmap();
        auto srcTensor = vkBn->getTensorBuffer(slice.origin);
        auto srcTensorSize = vkBn->getTensorSize(slice.origin);
        describe->writeBuffer(dstTensor.first->buffer(), 0, dstTensorSize, dstTensor.second);
        describe->writeBuffer(srcTensor.first->buffer(), 1, srcTensorSize, srcTensor.second);
        describe->writeBuffer(uniform->buffer(), 2, uniform->size());
        cmdBuffer->barrierSource(srcTensor.first->buffer(), srcTensor.second, srcTensorSize);
        blitPipeline->bind(cmdBuffer->get(), describe->get());
        vkCmdDispatch(cmdBuffer->get(), group, 1, 1);
        mExtraUniform.emplace_back(uniform);
        mExtraDescribes.emplace_back(describe);
    }
}


ErrorCode VulkanRaster::onEncode(const std::vector<Tensor *> &____inputs, const std::vector<Tensor *> &outputs,