Function main

tools/cpp/MNN2QNNModel.cpp:153–558

}

int main(int argc, const char* argv[]) {
    if (argc < 6) {
        MNN_PRINT("This tool generates offline caches for the QNN backend.");
        MNN_PRINT("Usage: %s <qnnSDKPath> <socId> <hexagonArch> <srcMNNPath> <outputDir> [totalShapeNum] [inputShape1] [inputShape2] ...\n", argv[0]);
        MNN_PRINT("    <qnnSDKPath>      : Path to the QNN SDK directory.\n");
        MNN_PRINT("    <socId>           : Target SoC ID.\n");
        MNN_PRINT("                        Common SoCs: 8Gen2 -> 43, 8Gen3 -> 57, 8 Elite -> 69. For others, please refer to Qualcomm's documentation.\n");
        MNN_PRINT("    <hexagonArch>     : Hexagon architecture version. This tool requires v73 or higher for weight sharing.\n");
        MNN_PRINT("                        Common Archs: 8Gen2 -> 73, 8Gen3 -> 75, 8 Elite -> 79. For others, please refer to Qualcomm's documentation.\n");
        MNN_PRINT("    <srcMNNPath>      : Path to the source MNN model file.\n");
        MNN_PRINT("    <outputDir>       : Directory to save the generated files, including a MNN model file with the suffix '.mnn' and a QNN serialized artifact with the suffix '.bin'.\n");
        MNN_PRINT("    [<totalShapeNum>] : Optional. Number of dynamic input shape configurations.\n");
        MNN_PRINT("    [<inputShapeN>]   : Optional. Input shape configuration. Can be a shape string or a path to a .mnn file.\n");
        MNN_PRINT("                     Shape string format for multiple inputs: dim1xdim2_dim3xdim4. Example: 1x3x512x512_1x256\n");
        MNN_PRINT("Examples:\n");
        MNN_PRINT("  1. Use default shape from the MNN model:\n");
        MNN_PRINT("     %s /path/to/qnn/sdk 57 75 /path/to/model.mnn /path/to/output\n", argv[0]);
        MNN_PRINT("  2. Specify two dynamic input shapes:\n");
        MNN_PRINT("     %s /path/to/qnn/sdk 57 75 /path/to/model.mnn /path/to/output 2 1x3x512x512_1x256 1x3x256x256_1x128\n", argv[0]);
        MNN_PRINT("     %s /path/to/qnn/sdk 57 75 /path/to/model.mnn /path/to/output 2 input_0.mnn input_1.mnn\n", argv[0]);

        return 1;
    }

    if (!checkSystem()) {
        return -1;
    }

    std::string qnnSdkPath = argv[1];
    int socId = std::stoi(std::string(argv[2]));
    int hexagonArch = std::stoi(std::string(argv[3]));
    const char* srcMNNPath = argv[4];
    std::string modelBaseName = [](const std::string& path) {
        std::string filename = path;
        auto pos = path.find_last_of("/\\");
        if (pos != std::string::npos) {
            filename = path.substr(pos + 1);
        }
        pos = filename.find_last_of('.');
        if (pos != std::string::npos) {
            return filename.substr(0, pos);
        }
        return filename;
    }(srcMNNPath);
    std::string modelSignature = "_" + std::to_string(socId) + "_" + std::to_string(hexagonArch);
    std::string outputDir = argv[5];
    std::string dstMNNPath = MNNFilePathConcat(outputDir, modelBaseName + modelSignature + ".mnn");

    std::vector<std::string> inputNames;
    std::vector<std::string> outputNames;
    std::vector<MNN::Express::VARP> inputs;
    std::vector<MNN::Express::VARP> outputs;
    std::vector<std::vector<std::vector<int>>> inputShapeLists;
    bool hasInputsVarp = false;
    std::vector<std::vector<MNN::Express::VARP>> inputsVarpList;

    int totalShapeType = 1;
    if(argc > 6) {