// hh_bench.sy -- Benchmark H-H gating speed.
// Run 1000 gate steps to measure throughput.

fn main() {
    let x = rand_tensor(1, 64)
    let w_inf = rand_tensor(64, 64)
    let b_inf = zeros(1, 64)
    let w_tau = rand_tensor(64, 64)
    let b_tau = zeros(1, 64)
    let w_h_inf = rand_tensor(64, 64)
    let b_h_inf = zeros(1, 64)
    let w_h_tau = rand_tensor(64, 64)
    let b_h_tau = zeros(1, 64)
    let m_state = ones(1, 64)
    let h_state = ones(1, 64)

    for i in 0..1000 {
        let result = hh_gate_full(x, w_inf, b_inf, w_tau, b_tau, w_h_inf, b_h_inf, w_h_tau, b_h_tau, m_state, h_state, 1.0)
    }
    println("1000 H-H gate steps done")
}
