Method onEncode

source/backend/vulkan/buffer/execution/VulkanRaster.cpp:116–346

ErrorCode VulkanRaster::onEncode(const std::vector<Tensor *> &____inputs, const std::vector<Tensor *> &outputs,
                           const VulkanCommandPool::Buffer *cmdBuffer) {
    MNN_ASSERT(outputs.size() == 1);
    if (____inputs.size() > 0) {
        OpCommonUtils::rasterInputReset(____inputs, outputs[0]);
    }
    auto output = outputs[0];
    auto des = TensorUtils::getDescribe(output);
    bool needZero = !TensorUtils::regionIsFull(output);
    _recycle();
    auto vkBn = static_cast<VulkanBackend*>(backend());
    auto bufferAlloc = vkBn->getDynamicMemoryPool();
    auto vkRt = ((VulkanRuntime*)(vkBn->getRuntime()));
    if (des->dimensionFormat == MNN_DATA_FORMAT_NC4HW4) {
        bool fast = true;
        for (int i=0; i< des->regions.size(); ++i) {
            auto& slice = des->regions[i];
            if (TensorUtils::getDescribe(slice.origin)->dimensionFormat != MNN_DATA_FORMAT_NC4HW4) {
                fast = false;
                break;
            }
            if (!OpCommonUtils::canBlitFast(slice, output, 4, true)) {
                fast = false;
                break;
            }
        }
        if (fast) {
            onEncodeFast(output, output, cmdBuffer, needZero);
            return NO_ERROR;
        }
    }
    // Single Convert Optimize
    std::vector<VkDescriptorType> nchwConvertTypes{VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
                                        VK_DESCRIPTOR_TYPE_STORAGE_BUFFER, VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER};
    do {
        if (des->regions.size() != 1) {
            break;
        }
        OpCommonUtils::TensorConvertParameter convertParameter;
        OpCommonUtils::turnRegion2Convert(des->regions[0], output, convertParameter);
        if (convertParameter.type == 0) {
            break;
        }
        const VulkanPipeline* convertPipeline = nullptr;
        std::string pKey = "glsl_";
        int srcIndex;
        int dstIndex;
        if (des->dimensionFormat == MNN_DATA_FORMAT_NC4HW4) {
            pKey += "nchwTonc4hw4_";
            srcIndex = 0;
            dstIndex = 1;
        } else {
            pKey += "nc4hw4Tonchw_";
            srcIndex = 1;
            dstIndex = 0;
        }
        if (output->getType().code == halide_type_float && vkBn->useFP16()) {
            pKey += "FP16_";