Method getRasterCacheCreateRecursive

source/geometry/GeometryComputer.cpp:283–370

}

void GeometryComputer::Context::getRasterCacheCreateRecursive(Tensor* src, CommandBuffer& cmd) {
    auto srcDes = TensorUtils::getDescribe(src);
    if (!_virtualMemory(srcDes)) {
        return;
    }
    if (_hasZeroDim(src)) {
        return;
    }
    bool needDelete = false;
    bool supportFuse = support(Interpreter::GEOMETRCOMPUTEMASK_FUSEREGION);
    bool supportFuseMulti = support(Interpreter::GEOMETRCOMPUTEMASK_FUSEREGION_MULTI);
    for (int regIndex = 0; regIndex < srcDes->regions.size();) {
        auto input = srcDes->regions.data() + regIndex;
        MNN_ASSERT(input->origin != src);
        
        auto inputDes = TensorUtils::getDescribe(input->origin);
        while (_virtualMemory(inputDes) && supportFuse) {
            if (0 == inputDes->regions.size()) {
                // Empty Input, Remove the region by set size as 0
                input->size[0] = 0;
                needDelete = true;
                break;
            }
            if (1 < inputDes->regions.size()) {
                if (!supportFuseMulti) {
                    break;
                }
                bool allCanMerge = true;
                for (auto& reg : inputDes->regions) {
                    allCanMerge = allCanMerge && mFuseUtils.match(reg, *input);
                    if (!allCanMerge) {
                        break;
                    }
                }
                if (!allCanMerge) {
                    break;
                }
                Tensor::InsideDescribe::Region backup = *input;
                mFuseUtils.match(inputDes->regions[0], *input);
                mFuseUtils.apply(inputDes->regions[0], *input);
                for (int i=1; i<inputDes->regions.size(); ++i) {
                    auto newReg = backup;
                    mFuseUtils.match(inputDes->regions[i], newReg);
                    mFuseUtils.apply(inputDes->regions[i], newReg);
                    if (newReg.size[0] == 0) {
                        continue;
                    }
                    srcDes->regions.emplace_back(newReg);
                }
                // After emplace_back, the input will change, reref it
                input = srcDes->regions.data() + regIndex;
                if (input->size[0] == 0) {
                    needDelete = true;
                    break;
                }
                inputDes = TensorUtils::getDescribe(input->origin);
                continue;
            }