Function test_graph_save_load_roundtrip_execution

tests/test_graph.cpp:738–791

}

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

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

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

        original.set_input(input_a, data_a.data(), Precision::FP16);
        original.set_input(input_b, data_b.data(), Precision::FP16);
        original.execute();

        __fp16* original_output = static_cast<__fp16*>(original.get_output(pow_id));
        std::vector<float> expected(6);
        for (size_t i = 0; i < expected.size(); ++i) {
            expected[i] = static_cast<float>(original_output[i]);
        }

        original.save(filename);

        CactusGraph loaded = CactusGraph::load(filename);
        GraphFile::SerializedGraph loaded_graph = GraphFile::load_graph(filename);
        const size_t loaded_input_a = runtime_id_from_serialized_index(loaded_graph.graph_inputs[0]);
        const size_t loaded_input_b = runtime_id_from_serialized_index(loaded_graph.graph_inputs[1]);
        const size_t loaded_output_id = runtime_id_from_serialized_index(loaded_graph.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* loaded_output = static_cast<__fp16*>(loaded.get_output(loaded_output_id));
        for (size_t i = 0; i < expected.size(); ++i) {
            float got = static_cast<float>(loaded_output[i]);
            if (std::abs(got - expected[i]) > 1e-3f) {
                std::cout << "[graph_save_load_roundtrip] mismatch at index " << i
                          << ": got=" << got
                          << " expected(original)=" << expected[i] << std::endl;
                std::remove(filename.c_str());
                return false;
            }
        }

        std::remove(filename.c_str());
        return true;
    } catch (const std::exception& e) {
        std::cout << "[graph_save_load_roundtrip] exception: " << e.what() << std::endl;
        return false;
    }
}

bool test_graph_save_for_inspection() {
    try {