Function test_node_save_load

tests/test_graph.cpp:609–655

}

bool test_node_save_load() {
    try {
        CactusGraph graph;

        size_t input_a = graph.input({2, 3}, Precision::FP16);
        size_t input_b = graph.input({2, 3}, Precision::FP16);
        size_t result_id = graph.add(input_a, input_b);

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

        graph.set_input(input_a, const_cast<void*>(static_cast<const void*>(data_a.data())), Precision::FP16);
        graph.set_input(input_b, const_cast<void*>(static_cast<const void*>(data_b.data())), Precision::FP16);
        graph.execute();

        std::string filename = "test_graph_save_load.bin";
        GraphFile::save_node(graph, result_id, filename);

        CactusGraph new_graph;
        size_t loaded_id = new_graph.mmap_weights(filename);
        new_graph.execute();

        __fp16* original_data = static_cast<__fp16*>(graph.get_output(result_id));
        __fp16* loaded_data = static_cast<__fp16*>(new_graph.get_output(loaded_id));

        for (size_t i = 0; i < 6; ++i) {
            if (std::abs(static_cast<float>(original_data[i]) - static_cast<float>(loaded_data[i])) > 1e-3f) {
                graph.hard_reset();
                new_graph.hard_reset();
                std::remove(filename.c_str());
                return false;
            }
        }

        const auto& buf = new_graph.get_output_buffer(loaded_id);
        bool result = (buf.shape == std::vector<size_t>{2, 3}) &&
                     (buf.precision == Precision::FP16) &&
                     (buf.byte_size == 12);

        graph.hard_reset();
        new_graph.hard_reset();
        std::remove(filename.c_str());
        return result;
    } catch (const std::exception& e) {
        return false;
    }
}


bool test_graph_save_load() {