Method onEncode

source/backend/vulkan/buffer/render/VulkanGaussianRender.cpp:54–186

    delete mContent;
}
ErrorCode VulkanRasterSort::onEncode(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs,
                                       const VulkanCommandPool::Buffer *cmdBuffer) {
    auto attr = inputs[0];
    auto numAttr = attr->length(1);
    auto extra = static_cast<VulkanBackend*>(backend());
    {
        std::vector<uint32_t> spc = {(uint32_t)mLocalSize, 1, 1, (uint32_t)mLocalSize};
        std::vector<VkDescriptorType> types {
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER
        };
        if (4 == numAttr) {
            mContent->rastersort_count_valid_number = extra->getPipelineFactory()->getPipeline("glsl_rastersort_count_valid_number_comp", types, spc);
        } else {
            mContent->rastersort_count_valid_number = extra->getPipelineFactory()->getPipeline("glsl_rastersort_count_valid_number_USE_HALF_comp", types, spc);
        }
    }
    {
        std::vector<uint32_t> spc = {(uint32_t)mLocalSize, 1, 1, (uint32_t)mLocalSize};
        std::vector<VkDescriptorType> types {
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER,
        };
        if (4 == numAttr) {
            mContent->rastersort_collect_key = extra->getPipelineFactory()->getPipeline("glsl_rastersort_collect_key_comp", types, spc);
        } else {
            mContent->rastersort_collect_key = extra->getPipelineFactory()->getPipeline("glsl_rastersort_collect_key_USE_HALF_comp", types, spc);
        }
    }
    mContent->reset();
    if(extra->isSupportAutotune()){
        autoTune(inputs, outputs);
    }
    auto memPool = extra->getDynamicMemoryPool();
    auto viewProj = inputs[1];
    auto numberPoint = attr->length(0);
    auto pointOffsetBytes = mLocalSize * mGroupSize * sizeof(uint32_t);
    auto pointOffsets = memPool->alloc(pointOffsetBytes);
    std::shared_ptr<VulkanBuffer> imageConstant = extra->allocUniform(nullptr, sizeof(ImageConstant));
    {
        auto ptr = (ImageConstant*)imageConstant->map();
        ptr->point[0] = numberPoint;
        //ptr->point[1] = attr->length(1) / 4;
        imageConstant->unmap();
    }
    mContent->uniforms.emplace_back(imageConstant);
    // Compute valid pointNumber
    {
        SharedPtr<VulkanLayout::DescriptorSet> des = mContent->rastersort_count_valid_number->createSet();
        des->writeBuffer(((VulkanBuffer*)pointOffsets.first)->buffer(), 0, pointOffsetBytes, pointOffsets.second);
        des->writeBuffer(extra->getBuffer(attr), 1);
        des->writeBuffer(extra->getBuffer(viewProj), 2);
        des->writeBuffer(imageConstant->buffer(), 3, imageConstant->size());