Function test_embedding_from_file

tests/test_graph.cpp:1299–1342

}

bool test_embedding_from_file() {
    CactusGraph graph;

    std::vector<__fp16> embeddings_data = {
        1.0f, 5.0f, 9.0f,
        2.0f, 6.0f, 10.0f,
        3.0f, 7.0f, 11.0f,
        4.0f, 8.0f, 12.0f
    };

    size_t temp_embeddings = graph.input({4, 3}, Precision::FP16);
    graph.set_input(temp_embeddings, embeddings_data.data(), Precision::FP16);

    const std::string temp_file = "test_embedding.bin";
    GraphFile::save_node(graph, temp_embeddings, temp_file);

    graph.hard_reset();

    size_t indices = graph.input({3}, Precision::INT8);
    size_t embedded = graph.embedding(temp_file, indices);

    std::vector<int8_t> idx_data = {2, 0, 3};
    graph.set_input(indices, idx_data.data(), Precision::INT8);
    graph.execute();

    std::vector<__fp16> expected = {
        3.0f, 7.0f, 11.0f,
        1.0f, 5.0f, 9.0f,
        4.0f, 8.0f, 12.0f
    };

    __fp16* output = static_cast<__fp16*>(graph.get_output(embedded));
    bool passed = true;
    for (size_t i = 0; i < expected.size(); i++) {
        if (std::abs(static_cast<float>(output[i]) - static_cast<float>(expected[i])) > 0.01f) {
            passed = false;
            break;
        }
    }

    std::remove(temp_file.c_str());

    return passed;
}

bool test_stft() {
    const size_t N = 2, C_in = 1, L = 8, K = 4, stride = 2, num_fft_bins = 2;