Function test_neon_reduction_correctness

tests/test_kernel.cpp:104–122

}

bool test_neon_reduction_correctness() {
    std::vector<__fp16> input = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f};

    double sum_result = cactus_sum_all_f16(input.data(), input.size());
    double expected_sum = 36.0;

    if (std::abs(sum_result - expected_sum) > 1e-2) {
        return false;
    }

    double mean_result = cactus_mean_all_f16(input.data(), input.size());
    double expected_mean = 4.5;

    if (std::abs(mean_result - expected_mean) > 1e-2) {
        return false;
    }

    return true;
}

bool test_neon_transpose_fp16_correctness() {
    const size_t M = 3, N = 4;