Function test_graph_save_load

tests/test_graph.cpp:658–736

bool test_graph_save_load() {
    try {
        const std::string filename = "test_graph_save_load.cg";

        CactusGraph graph;
        size_t input_a = graph.input({2, 3}, Precision::FP16);
        size_t input_b = graph.input({2, 3}, Precision::FP16);
        size_t sum_id = graph.add(input_a, input_b);
        graph.pow(sum_id, 2.0f);

        graph.save(filename);

        GraphFile::SerializedGraph sg = GraphFile::load_graph(filename);

        if (sg.header.node_count != 4) {
            std::cout << "[graph_save_load] unexpected node_count: "
                      << sg.header.node_count << " expected 4" << std::endl;
            std::remove(filename.c_str());
            return false;
        }

        if (sg.graph_inputs.size() != 2 || sg.graph_inputs[0] != 0 ||
  sg.graph_inputs[1] != 1) {
            std::cout << "[graph_save_load] unexpected graph_inputs:";
            for (uint32_t idx : sg.graph_inputs) {
                std::cout << " " << idx;
            }
            std::cout << std::endl;
            std::remove(filename.c_str());
            return false;
        }

        if (sg.graph_outputs.size() != 1 || sg.graph_outputs[0] != 3) {
            std::cout << "[graph_save_load] unexpected graph_outputs:";
            for (uint32_t idx : sg.graph_outputs) {
                std::cout << " " << idx;
            }
            std::cout << std::endl;
            std::remove(filename.c_str());
            return false;
        }

        CactusGraph loaded = CactusGraph::load(filename);

        std::vector<__fp16> data_a = {1, 2, 3, 4, 5, 6};
        std::vector<__fp16> data_b = {10, 20, 30, 40, 50, 60};

        const size_t loaded_input_a = runtime_id_from_serialized_index(sg.graph_inputs[0]);
        const size_t loaded_input_b = runtime_id_from_serialized_index(sg.graph_inputs[1]);
        const size_t loaded_output_id = runtime_id_from_serialized_index(sg.graph_outputs[0]);

        loaded.set_input(loaded_input_a, data_a.data(), Precision::FP16);
        loaded.set_input(loaded_input_b, data_b.data(), Precision::FP16);
        loaded.execute();

        __fp16* output = static_cast<__fp16*>(loaded.get_output(loaded_output_id));
        std::vector<float> expected = {
            121.0f, 484.0f, 1089.0f,