Function main

tests/test_e2e_pvs.cpp:99–333

}

int main(int argc, char ** argv) {
    if (argc < 3) {
        fprintf(stderr, "Usage: %s <model.ggml> <ref_dir>\n", argv[0]);
        fprintf(stderr, "\nref_dir: directory with Python reference tensors\n");
        fprintf(stderr, "  Should contain preprocessed.bin/.shape and phase6/cat_box/\n");
        return 1;
    }

    const std::string model_path = argv[1];
    const std::string ref_dir = argv[2];
    const std::string cpp_out = ref_dir + "/cpp_out";
    const std::string cpp_p6_out = ref_dir + "/cpp_out_phase6";
    ensure_dir(cpp_out);
    ensure_dir(cpp_p6_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");
    if (ref_img.data.empty()) {
        fprintf(stderr, "Failed to load %s/preprocessed.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");

    // ═══ Collect all metrics ═══
    std::vector<metric_row> report;

    // Helper: dump state tensor, compare with ref in NHWC layout
    auto check_nhwc = [&](const char * stage, const char * tensor_name,
                          const char * ref_name, float atol) {
        sam3_dump_state_tensor(*state, tensor_name, cpp_out + "/" + tensor_name);