Function main

tests/test_metal_python_ref_compare.cpp:468–503

}

int main(int argc, char ** argv) {
    if (argc < 2) {
        fprintf(stderr,
                "Usage: %s <model.ggml> [ref_dir] [backend=metal|cpu|both] [phase6_case_tsv]\n",
                argv[0]);
        fprintf(stderr,
                "Defaults: ref_dir=tests/debug_pipeline/e2e_ref backend=both phase6_case_tsv=tests/debug_pipeline/phase6_cat_box.tsv\n");
        return 1;
    }

    const std::string model_path = argv[1];
    const std::string ref_dir = argc >= 3 ? argv[2] : "tests/debug_pipeline/e2e_ref";
    const std::string backend_mode = argc >= 4 ? argv[3] : "both";
    const std::string case_tsv = argc >= 5 ? argv[4] : "tests/debug_pipeline/phase6_cat_box.tsv";

    sam3_pvs_params pvs_params;
    std::string phase6_case_id;
    if (!parse_phase6_case(case_tsv, pvs_params, phase6_case_id)) {
        fprintf(stderr, "Failed to parse phase 6 case file: %s\n", case_tsv.c_str());
        return 1;
    }

    std::vector<backend_run> runs;
    if (backend_mode == "cpu" || backend_mode == "both") {
        runs.push_back(run_backend("cpu", false, model_path, ref_dir, pvs_params, phase6_case_id));
    }
    if (backend_mode == "metal" || backend_mode == "both") {
        runs.push_back(run_backend("metal", true, model_path, ref_dir, pvs_params, phase6_case_id));
    }

    for (const auto & run : runs) {
        print_summary(run);
    }

    return 0;
}