Function main

tests/test_e2e_pcs.cpp:108–468

}

int main(int argc, char ** argv) {
    if (argc < 3) {
        fprintf(stderr, "Usage: %s <model.ggml> <ref_dir>\n", argv[0]);
        fprintf(stderr, "\nref_dir should contain pcs_ref/ from dump_e2e_pcs.py\n");
        return 1;
    }

    const std::string model_path = argv[1];
    const std::string ref_dir = std::string(argv[2]) + "/pcs_ref";
    const std::string cpp_out = std::string(argv[2]) + "/pcs_cpp";
    ensure_dir(cpp_out);

    // ═══ Load model ═══
    fprintf(stderr, "\n═══ Loading model ═══\n");
    sam3_params params;
    params.model_path = model_path;
    params.use_gpu = false;
    params.n_threads = 4;

    auto model = sam3_load_model(params);
    if (!model) {
        fprintf(stderr, "Failed to load model\n");
        return 1;
    }

    auto state = sam3_create_state(*model, params);
    if (!state) {
        fprintf(stderr, "Failed to create state\n");
        return 1;
    }

    // ═══ Stage 1: Load Python-preprocessed image ═══
    fprintf(stderr, "\n═══ Stage 1: Loading Python-preprocessed image ═══\n");
    auto ref_img = load_ref_f32(ref_dir + "/preprocessed_chw");
    if (ref_img.data.empty()) {
        fprintf(stderr, "Failed to load %s/preprocessed_chw.bin\n", ref_dir.c_str());
        return 1;
    }
    fprintf(stderr, "  Loaded preprocessed image: %d elements\n", ref_img.numel());

    const int img_size = 1008;
    bool ok = sam3_encode_image_from_preprocessed(*state, *model, ref_img.data.data(), img_size);
    if (!ok) {
        fprintf(stderr, "sam3_encode_image_from_preprocessed failed!\n");
        return 1;
    }
    fprintf(stderr, "  Image encoded successfully\n");

    // ═══ Stage 2: Dump ViT block outputs (from state debug tensors) ═══
    fprintf(stderr, "\n═══ Stage 2: ViT Backbone ═══\n");
    std::vector<metric_row> report;

    // ViT block outputs are stored in state during encoding
    for (int i : {0, 7, 15, 23, 31}) {
        char cpp_name[64], ref_name[64];
        snprintf(cpp_name, sizeof(cpp_name), "dbg_block_%d_out", i);
        snprintf(ref_name, sizeof(ref_name), "vit_block_%02d_out", i);