Function test_basic_operation

tests/test_utils.cpp:132–157

}

bool test_basic_operation(const std::string& op_name,
                          std::function<size_t(CactusGraph&, size_t, size_t)> op_func,
                          const std::vector<__fp16>& data_a,
                          const std::vector<__fp16>& data_b,
                          const std::vector<__fp16>& expected,
                          const std::vector<size_t>& shape) {
    (void)op_name;
    CactusGraph graph;
    size_t input_a = graph.input(shape, Precision::FP16);
    size_t input_b = graph.input(shape, Precision::FP16);
    size_t result_id = op_func(graph, input_a, input_b);

    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();

    __fp16* output = static_cast<__fp16*>(graph.get_output(result_id));
    for (size_t i = 0; i < expected.size(); ++i) {
        if (std::abs(static_cast<float>(output[i]) - static_cast<float>(expected[i])) > 1e-2f) {
            graph.hard_reset();
            return false;
        }
    }
    graph.hard_reset();
    return true;
}

bool test_scalar_operation(const std::string& op_name,
                           std::function<size_t(CactusGraph&, size_t, float)> op_func,