Function test_stft

tests/test_graph.cpp:1344–1387

}

bool test_stft() {
    const size_t N = 2, C_in = 1, L = 8, K = 4, stride = 2, num_fft_bins = 2;
    const size_t C_out = 2 * num_fft_bins;
    const size_t out_len = (L - K) / stride + 1;

    std::vector<__fp16> weight_data = {
        (__fp16) 1, (__fp16) 1, (__fp16) 1, (__fp16) 1,
        (__fp16) 1, (__fp16) 0, (__fp16)-1, (__fp16) 0,
        (__fp16) 0, (__fp16) 0, (__fp16) 0, (__fp16) 0,
        (__fp16) 0, (__fp16)-1, (__fp16) 0, (__fp16) 1,
    };
    std::vector<__fp16> input_data = {
        (__fp16)1, (__fp16)2, (__fp16)3, (__fp16)4, (__fp16)5, (__fp16)6, (__fp16)7, (__fp16)8,
        (__fp16)0, (__fp16)1, (__fp16)0, (__fp16)-1, (__fp16)0, (__fp16)1, (__fp16)0, (__fp16)-1,
    };

    TestUtils::FP16TestFixture fx;
    size_t inp = fx.create_input({N, C_in, L});
    size_t wt  = fx.create_input({C_out, C_in, K});
    size_t out = fx.graph().stft(inp, wt, stride, num_fft_bins);

    if (fx.graph().get_output_buffer(out).shape != std::vector<size_t>{N, C_out, out_len}) return false;

    fx.set_input_data(inp, input_data);
    fx.set_input_data(wt, weight_data);
    fx.execute();

    const __fp16* cplx = fx.get_output(out);
    const size_t out_bs = C_out * out_len;
    const float tol = 0.1f;

    for (size_t t = 0; t < out_len; ++t) {
        if (std::abs((float)cplx[1 * out_len + t] - (-2.0f)) > tol) return false;
        if (std::abs((float)cplx[(1 + num_fft_bins) * out_len + t] - 2.0f) > tol) return false;
    }

    const float batch1_bin1_imag[3] = {-2.0f, 2.0f, -2.0f};
    for (size_t t = 0; t < out_len; ++t) {
        if (std::abs((float)cplx[out_bs + 1 * out_len + t] - 0.0f) > tol) return false;
        if (std::abs((float)cplx[out_bs + (1 + num_fft_bins) * out_len + t] - batch1_bin1_imag[t]) > tol) return false;
    }

    return true;
}

template<typename T>
static bool run_layernorm_case(