Function main

kleidiai-examples/audiogen-et/app/main.cpp:231–492

}

int main(int32_t argc, char** argv) {

    // Required arguments
    std::string models_base_path = "";
    std::string prompt           = "";
    size_t cpu_threads           = -1;

    // Optional arguments
    std::string output_file      = "";
    size_t seed                  = k_seed_default;
    size_t num_steps             = k_num_steps_default;
    float audio_len_sec          = static_cast<float>(k_audio_len_sec_default);
    bool  run_dummy_run          = false;

    int32_t opt;
    while ((opt = getopt(argc, argv, "m:p:t:s:n:o:l:d:h")) != -1) {
        switch (opt) {
            case 'm': models_base_path = optarg; break;
            case 'p': prompt           = optarg; break;
            case 't': cpu_threads      = std::stoull(optarg); break;
            case 'o': output_file      = optarg; break;
            case 's': seed             = std::stoull(optarg); break;
            case 'n': num_steps        = std::stoull(optarg); break;
            case 'l': audio_len_sec    = static_cast<float>(std::stoull(optarg)); break;
            case 'd': run_dummy_run    = (std::string(optarg) == "true"); break;
            case 'h':
            default:
                print_usage(argv[0]);
                return EXIT_FAILURE;
        }
    }

    // Check the mandatory arguments
    if (models_base_path.empty() || prompt.empty() || cpu_threads <= 0) {
        fprintf(stderr, "ERROR: Missing required arguments.\n\n");
        print_usage(argv[0]);
        return EXIT_FAILURE;
    }

    std::string t5_model = models_base_path + "/conditioners_model.pte";
    std::string dit_model = models_base_path + "/dit_model.pte";
    std::string autoencoder_model = models_base_path + "/autoencoder_model.pte";
    std::string sentence_model_path = models_base_path + "/spiece.model";

#if defined(ET_USE_THREADPOOL)
    uint32_t num_performant_cores = cpu_threads == -1
      ? ::executorch::extension::cpuinfo::get_num_performant_cores()
      : static_cast<uint32_t>(cpu_threads);
    ET_LOG(
      Info, "Resetting threadpool with num threads = %d", num_performant_cores);
    if (num_performant_cores > 0) {
    ::executorch::extension::threadpool::get_threadpool()
        ->_unsafe_reset_threadpool(num_performant_cores);
    }
#else
    uint32_t num_performant_cores = 4;
#endif
    ET_LOG(Info, "Using %d threads", num_performant_cores);