Function main

tests/test_phase4.cpp:150–334

}

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

    const std::string ref_dir = argv[1];
    const std::string model_path = argv[2];
    const std::string prompts_path = argv[3];
    const float atol = 1e-4f;

    auto prompts = load_prompt_cases(prompts_path);
    if (prompts.empty()) {
        fprintf(stderr, "No prompts loaded from %s\n", prompts_path.c_str());
        return 1;
    }

    if (!sam3_test_load_tokenizer(model_path)) {
        fprintf(stderr, "Failed to load tokenizer from %s\n", model_path.c_str());
        return 1;
    }

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

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

    const std::string cpp_root = ref_dir + "/cpp_out_phase4";
    if (!ensure_dir(cpp_root)) {
        fprintf(stderr, "Failed to create %s\n", cpp_root.c_str());
        return 1;
    }

    const std::vector<std::string> names = tensor_names();
    const std::vector<std::string> category_order = {
        "token_ids",
        "causal_mask",
        "text_token_embed",
        "text_after_pos_embed",
        "block_after_ln1",
        "block_qkv",
        "block_attn_out",
        "block_after_attn_residual",
        "block_after_ln2",
        "block_mlp_fc1",
        "block_mlp_gelu",
        "block_mlp_out",
        "block_out",
        "text_final_ln",
        "text_features_2d",
    };