Function main

tools/cpp/compilefornpu.cpp:1450–1857

}

int main(int argc, const char* argv[]) {
    if (argc < 3) {
        MNN_PRINT("Usage: ./compilefornpu src.mnn dst.mnn npu.json\n");
        return 0;
    }
    const char* srcMNN = argv[1];
    const char* dstMNN = argv[2];
    std::vector<std::string> inputNames;
    std::vector<std::string> outputNames;
    std::vector<std::vector<MNN::Express::VARP>> inputs;
    std::set<std::string> skipOps;

    if (argc >= 4) {
        rapidjson::Document document;
        std::ifstream fileNames(argv[3]);
        std::ostringstream output;
        output << fileNames.rdbuf();
        auto outputStr = output.str();
        document.Parse(outputStr.c_str());
        if (document.HasParseError()) {
            MNN_ERROR("Invalid json\n");
            return 0;
        }
        gNPUName = document["type"].GetString();
        if (gNPUName == "QNN") {
            MNN_PRINT("Convert for QNN, QualComn's NPU\n");
            gNPUType = MNN_CONVERT_QNN;
            gNeedOffline = true;
            gOfflieSrc = "";
            gOfflieDst = "bin";
        } else if (gNPUName == "MLDA") {
            MNN_PRINT("Convert for MLDA, MTK's NPU\n");
            gNPUType = MNN_CONVERT_NEUROPILOT;
            gNeedOffline = true;
            gOfflieSrc = "tflite";
            gOfflieDst = "dla";
        } else if (gNPUName == "CoreML") {
            MNN_PRINT("Convert for CoreML, Apple's framework\n");
            gNPUType = MNN_CONVERT_COREML;
            gNeedOffline = true;
            gOfflieSrc = "";
            gOfflieDst = "";
        } else {
            MNN_PRINT("Use Native NPU compute\n");
        }
        if (document.HasMember("cache")) {
            gCacheDir = document["cache"].GetString();
            FUNC_PRINT_ALL(gCacheDir.c_str(), s);
            MNNCreateDir(gCacheDir.c_str());
        }
        if (document.HasMember("graph_name")) {
            gGraphName = document["graph_name"].GetString();
        }
        if (document.HasMember("skips")) {
            auto skips = document["skips"].GetArray();
            for (auto iter = skips.Begin(); iter != skips.End(); iter++) {
                skipOps.insert(iter->GetString());
            }