Function main

tests/test_metal_phase6.cpp:115–195

}

int main(int argc, char ** argv) {
    if (argc < 5) {
        fprintf(stderr, "Usage: %s <ref_dir> <prephase_ref_dir> <model_path> <cases_tsv>\n", argv[0]);
        return 1;
    }

    const std::string ref_dir = argv[1];
    const std::string prephase_ref_dir = argv[2];
    const std::string model_path = argv[3];
    const std::string cases_path = argv[4];
    const std::string cpp_root = ref_dir + "/cpp_out_phase6_metal";

    auto cases = load_cases(cases_path);
    if (cases.empty()) {
        fprintf(stderr, "No phase 6 cases loaded from %s\n", cases_path.c_str());
        return 1;
    }

    if (!ensure_dir(cpp_root)) {
        fprintf(stderr, "Failed to create %s\n", cpp_root.c_str());
        return 1;
    }

    sam3_params params;
    params.model_path = model_path;
    params.n_threads = 1;
    params.use_gpu = true;

    auto model = sam3_load_model(params);
    if (!model) {
        fprintf(stderr, "Failed to load model from %s\n", model_path.c_str());
        return 1;
    }

    bool overall_ok = true;
    for (const auto & tc : cases) {
        fprintf(stderr, "\n=== Metal Phase 6 Case %s ===\n", tc.id.c_str());

        const std::string case_ref_dir = ref_dir + "/" + tc.id;
        const std::string case_cpp_dir = cpp_root + "/" + tc.id;
        if (!ensure_dir(case_cpp_dir)) {
            fprintf(stderr, "Failed to create %s\n", case_cpp_dir.c_str());
            overall_ok = false;
            continue;
        }

        if (!sam3_test_dump_phase6_from_ref_inputs(*model, prephase_ref_dir, tc.params,
                                                   case_cpp_dir, params.n_threads)) {
            fprintf(stderr, "  [FAIL] could not run Metal phase 6 dump for %s\n", tc.id.c_str());
            overall_ok = false;
            continue;
        }

        for (const auto & tensor : tensor_cases()) {
            auto ref = load_ref_f32(case_ref_dir + "/" + tensor.name);
            auto got = load_ref_f32(case_cpp_dir + "/" + tensor.name);
            if (ref.data.empty() || got.data.empty() || ref.numel() != got.numel()) {
                fprintf(stderr, "  [FAIL] %s missing or shape mismatch\n", tensor.name);